Spaces:

gelnesr
/

Dyna-1

Running on Zero

gelnesr commited on Mar 26

Commit

514b274

verified ·

1 Parent(s): e80669e

update device

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ from huggingface_hub import get_hf_file_metadata, hf_hub_download, login
 import spaces
 import gradio as gr
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 REPO_URL = "https://github.com/WaymentSteeleLab/Dyna-1.git"
 DYNA_MODEL_ID = "gelnesr/Dyna-1"

 import spaces
 import gradio as gr
+DEVICE = torch.device('cpu')
 REPO_URL = "https://github.com/WaymentSteeleLab/Dyna-1.git"
 DYNA_MODEL_ID = "gelnesr/Dyna-1"