Spaces:

GarGerry
/

ChatBotAi

Sleeping

GarGerry commited on Jan 14

Commit

a28907d

verified ·

1 Parent(s): 6cdfbea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-128k-instruct",
-    device_map="cuda",  # Gunakan 'cpu' jika tidak menggunakan GPU
     torch_dtype="auto",
     trust_remote_code=True,
     attn_implementation="eager"  # Menggunakan eager untuk menghindari masalah flash-attention

 torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-128k-instruct",
+    device_map="cpu",  # Gunakan 'cpu' jika tidak ada GPU
     torch_dtype="auto",
     trust_remote_code=True,
     attn_implementation="eager"  # Menggunakan eager untuk menghindari masalah flash-attention