Skip to content

Commit 2544648

Browse files
committed
upd docs a little bit
1 parent ee71fcd commit 2544648

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/autointent/modules/scoring/_ptuning/ptuning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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,

user_guides/basic_usage/03_automl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@
118118
from 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)
124124
custom_pipeline.set_config(

0 commit comments

Comments
 (0)