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", AVAILABLE_MODELS.get("multilingual-e5-base", "intfloat/multilingual-e5-base"))
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