Spaces:

izuemon
/

phi-3

Running

izuemon commited on Mar 15

Commit

97dbf57

verified ·

1 Parent(s): 229ea4d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-4k-instruct",
-    device_map="cuda",
     torch_dtype="auto",
     trust_remote_code=True
 )

 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-4k-instruct",
+    device_map="cpu",
     torch_dtype="auto",
     trust_remote_code=True
 )