Davidtran99
commited on
Commit
·
eef3d60
1
Parent(s):
493d90f
fix(embeddings): remove fallback for BGE-M3
Browse files
backend/hue_portal/core/embeddings.py
CHANGED
|
@@ -46,7 +46,7 @@ AVAILABLE_MODELS = {
|
|
| 46 |
# - EMBEDDING_MODEL=username/private-model (private HF model, requires HF_TOKEN)
|
| 47 |
DEFAULT_MODEL_NAME = os.environ.get(
|
| 48 |
"EMBEDDING_MODEL",
|
| 49 |
-
AVAILABLE_MODELS.get("bge-m3",
|
| 50 |
)
|
| 51 |
FALLBACK_MODEL_NAME = AVAILABLE_MODELS.get("paraphrase-multilingual", "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
|
| 52 |
|
|
|
|
| 46 |
# - EMBEDDING_MODEL=username/private-model (private HF model, requires HF_TOKEN)
|
| 47 |
DEFAULT_MODEL_NAME = os.environ.get(
|
| 48 |
"EMBEDDING_MODEL",
|
| 49 |
+
AVAILABLE_MODELS.get("bge-m3", "BAAI/bge-m3") # BGE-M3 is default, no fallback
|
| 50 |
)
|
| 51 |
FALLBACK_MODEL_NAME = AVAILABLE_MODELS.get("paraphrase-multilingual", "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
|
| 52 |
|