File tree Expand file tree Collapse file tree
src/autointent/modules/scoring/_ptuning Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -35,7 +35,7 @@ class PTuningScorer(BertScorer):
3535
3636 from autointent.modules import PTuningScorer
3737 scorer = PTuningScorer(
38- classification_model_config="prajjwal1/bert-tiny ",
38+ classification_model_config="cointegrated/rubert-tiny2 ",
3939 num_train_epochs=3,
4040 batch_size=8,
4141 num_virtual_tokens=10,
Original file line number Diff line number Diff line change 118118from autointent .configs import CrossEncoderConfig , TokenizerConfig , get_default_embedder_config
119119
120120# Configure embedding model (used for vector representations)
121- custom_pipeline .set_config (get_default_embedder_config (model_name = "prajjwal1/bert-tiny " ))
121+ custom_pipeline .set_config (get_default_embedder_config (model_name = "cointegrated/rubert-tiny2 " ))
122122
123123# Configure cross-encoder model (used for scoring text pairs)
124124custom_pipeline .set_config (
You can’t perform that action at this time.
0 commit comments