Changes test to BAAI/bge-base-en-v1.5

Signed-off-by: Flavia Beo <[email protected]>
flaviabeo · Oct 23, 2024 · caa731a · caa731a
1 parent 337394e
commit caa731a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/model_executor/test_model_load_with_params.py b/tests/model_executor/test_model_load_with_params.py
@@ -2,7 +2,7 @@
 
 MAX_MODEL_LEN = 128
 MODEL_NAME = os.environ.get("MODEL_NAME",
-                            "sentence-transformers/all-MiniLM-L12-v2")
+                            "BAAI/bge-base-en-v1.5")
 REVISION = os.environ.get("REVISION", "main")