J266501 commited on
Commit
fd4cc6e
·
verified ·
1 Parent(s): fc02a5e
Files changed (1) hide show
  1. api_service.py +1 -1
api_service.py CHANGED
@@ -28,7 +28,7 @@ VECTOR_DB_DIR = os.getenv("VECTOR_DB_DIR", "/tmp/vector_db_chroma")
28
  COLLECTION_NAME = "my_instrument_manual_chunks"
29
 
30
  LLM_MODEL_NAME = "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"
31
- EMBEDDINGS_MODEL_NAME = "togethercomputer/m2-bert-80M-32k-retrieval"
32
 
33
  # --- 2. Lifespan Event Handler ---
34
  @asynccontextmanager
 
28
  COLLECTION_NAME = "my_instrument_manual_chunks"
29
 
30
  LLM_MODEL_NAME = "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"
31
+ EMBEDDINGS_MODEL_NAME = "BAAI/bge-base-en-v1.5"
32
 
33
  # --- 2. Lifespan Event Handler ---
34
  @asynccontextmanager