Johnyquest7 commited on
Commit
3e8e7f4
1 Parent(s): 2961e7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ CSS = """
103
 
104
 
105
  tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
106
- model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", torch_dtype="auto", device_map="auto")
107
 
108
 
109
  @spaces.GPU
 
103
 
104
 
105
  tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-4k-instruct")
106
+ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-4k-instruct", torch_dtype="auto", device_map="auto",trust_remote_code=True)
107
 
108
 
109
  @spaces.GPU