J266501 commited on
Commit
fc02a5e
·
verified ·
1 Parent(s): be91d4f
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -42,7 +42,7 @@ except Exception as e:
42
  print("Please check your TOGETHER_API_KEY and model name.")
43
 
44
  # --- Retriever Setup ---
45
- embeddings_model_name = "togethercomputer/m2-bert-80M-32k-retrieval" # 必須與 setup_knowledge_base.py 中使用的模型一致
46
 
47
  try:
48
  retriever_embeddings = TogetherEmbeddings(
 
42
  print("Please check your TOGETHER_API_KEY and model name.")
43
 
44
  # --- Retriever Setup ---
45
+ embeddings_model_name = "BAAI/bge-base-en-v1.5" # 必須與 setup_knowledge_base.py 中使用的模型一致
46
 
47
  try:
48
  retriever_embeddings = TogetherEmbeddings(