eaglelandsonce commited on
Commit
efaa4a0
β€’
1 Parent(s): 31f3df3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +34,8 @@ QA_PROMPT = PromptTemplate(
34
  )
35
 
36
  # Load Phi-2 model from hugging face hub
37
- # model_id = "microsoft/phi-2"
38
- model_id = "HuggingFaceH4/zephyr-7b-alpha"
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)
 
34
  )
35
 
36
  # Load Phi-2 model from hugging face hub
37
+ model_id = "microsoft/phi-2"
38
+
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)