Hugging Face's logo

Spaces:
Khoa
/
phi-2-example
Runtime error

App Files Community
1
Khoa commited on
Commit
8926d05
·
verified ·
1 Parent(s): da54301

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
-
 
3
  torch.set_default_device("cuda")
4
 
5
  model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True)
 
1
  import gradio as gr
2
+ import torch
3
+ from transformers import AutoModelForCausalLM, AutoTokenizer
4
  torch.set_default_device("cuda")
5
 
6
  model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype="auto", trust_remote_code=True)