Spaces:

shrimantasatpati
/

Microsoft_Phi2_Streamlit

Runtime error

shrimantasatpati commited on Dec 16, 2023

Commit

3b4796e

•

1 Parent(s): 18e4bcd

Updated app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/phi-2",
     device_map="auto",
-    trust_remote_code=True
     offload_folder="offload"
 )

 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/phi-2",
     device_map="auto",
+    trust_remote_code=True,
     offload_folder="offload"
 )