Hugging Face's logo

Spaces:
eaglelandsonce
/
RAG-with-Phi-2-and-LangChain
Runtime error

App Files Community
eaglelandsonce commited on
Commit
4530c41
·
verified ·
1 Parent(s): f0a4462

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,8 @@ QA_PROMPT = PromptTemplate(
34
  )
35
 
36
  # Load Phi-2 model from hugging face hub
37
- model_id = "microsoft/phi-2"
 
38
 
39
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
40
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32, device_map="auto", trust_remote_code=True)
 
34
  )
35
 
36
  # Load Phi-2 model from hugging face hub
37
+ # model_id = "microsoft/phi-2"
38
+ model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
39
 
40
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
41
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float32, device_map="auto", trust_remote_code=True)