Hugging Face's logo

Spaces:
gelnesr
/
Dyna-1
Running on Zero

App Files Community
gelnesr commited on
Commit
514b274
·
verified ·
1 Parent(s): e80669e

update device

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ from huggingface_hub import get_hf_file_metadata, hf_hub_download, login
22
  import spaces
23
  import gradio as gr
24
 
25
- DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
26
  REPO_URL = "https://github.com/WaymentSteeleLab/Dyna-1.git"
27
  DYNA_MODEL_ID = "gelnesr/Dyna-1"
28
 
 
22
  import spaces
23
  import gradio as gr
24
 
25
+ DEVICE = torch.device('cpu')
26
  REPO_URL = "https://github.com/WaymentSteeleLab/Dyna-1.git"
27
  DYNA_MODEL_ID = "gelnesr/Dyna-1"
28