diff --git a/autointent/_datafiles/default-multiclass-config.yaml b/autointent/_datafiles/default-multiclass-config.yaml index d38074ac3..96c74057b 100644 --- a/autointent/_datafiles/default-multiclass-config.yaml +++ b/autointent/_datafiles/default-multiclass-config.yaml @@ -6,7 +6,7 @@ k: [10] embedder_name: - avsolatorio/GIST-small-Embedding-v0 - - infgrad/stella-base-en-v2 + - sergeyzh/rubert-tiny-turbo - node_type: scoring target_metric: scoring_roc_auc search_space: @@ -16,7 +16,6 @@ - module_name: linear - module_name: dnnc cross_encoder_name: - - BAAI/bge-reranker-base - cross-encoder/ms-marco-MiniLM-L-6-v2 k: [1, 3, 5, 10] - node_type: decision diff --git a/autointent/_datafiles/inference-config-example.yaml b/autointent/_datafiles/inference-config-example.yaml index 1b4852009..ef81188ff 100644 --- a/autointent/_datafiles/inference-config-example.yaml +++ b/autointent/_datafiles/inference-config-example.yaml @@ -2,7 +2,7 @@ module_name: retrieval module_config: k: 10 - model_name: infgrad/stella-base-en-v2 + model_name: sergeyzh/rubert-tiny-turbo load_path: . - node_type: scoring module_name: knn diff --git a/user_guides/advanced/02_search_space.py b/user_guides/advanced/02_search_space.py index 032d8af4e..c2531cd94 100644 --- a/user_guides/advanced/02_search_space.py +++ b/user_guides/advanced/02_search_space.py @@ -24,7 +24,7 @@ knn_module = { "module_name": "knn", "k": [1, 5, 10, 50], - "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"], + "embedder_name": ["sergeyzh/rubert-tiny-turbo"], } # %% [markdown] @@ -78,7 +78,7 @@ { "module_name": "retrieval", "k": [10], - "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"], + "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"], } ], }, @@ -90,7 +90,7 @@ {"module_name": "linear"}, { "module_name": "dnnc", - "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"], + "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"], "k": [1, 3, 5, 10], }, ], diff --git a/user_guides/advanced/04_reporting.py b/user_guides/advanced/04_reporting.py index 2f04f70a8..868153d55 100644 --- a/user_guides/advanced/04_reporting.py +++ b/user_guides/advanced/04_reporting.py @@ -14,7 +14,7 @@ { "module_name": "retrieval", "k": [10], - "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "infgrad/stella-base-en-v2"], + "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"], } ], }, @@ -26,7 +26,7 @@ {"module_name": "linear"}, { "module_name": "dnnc", - "cross_encoder_name": ["BAAI/bge-reranker-base", "cross-encoder/ms-marco-MiniLM-L-6-v2"], + "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"], "k": [1, 3, 5, 10], }, ],