kanninian commited on
Commit
92ac835
·
verified ·
1 Parent(s): 55fdddf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,10 +11,10 @@ from qa_vector_store import build_qa_vector_store, retrieve_and_rerank, generate
11
  model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
12
  collection_name = model_name.split("/")[-1]
13
  cross_encoder_model = CrossEncoder("cross-encoder/mmarco-mMiniLMv2-L12-H384-v1")
14
- # tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
15
- # llm_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
16
- tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
17
- llm_model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", trust_remote_code=True)
18
 
19
  # 構建向量資料庫
20
  build_qa_vector_store(model_name, collection_name)
 
11
  model_name = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
12
  collection_name = model_name.split("/")[-1]
13
  cross_encoder_model = CrossEncoder("cross-encoder/mmarco-mMiniLMv2-L12-H384-v1")
14
+ tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
15
+ llm_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B-Instruct", trust_remote_code=True)
16
+ # tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
17
+ # llm_model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", trust_remote_code=True)
18
 
19
  # 構建向量資料庫
20
  build_qa_vector_store(model_name, collection_name)