Spaces:

huolongguo10
/

microsoft-phi-2

Runtime error

huolongguo10 commited on Dec 14, 2023

Commit

517a7f0

•

1 Parent(s): 5df8faa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 # torch.set_default_device("cuda")
 model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype=torch.float32, device_map="cpu", trust_remote_code=True)

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
 # torch.set_default_device("cuda")
 model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype=torch.float32, device_map="cpu", trust_remote_code=True)