Hugging Face's logo

Spaces:
cosmicdream
/
Image_Variations
Runtime error

App Files Community
2
cosmicdream commited on
Commit
1321dbf
·
1 Parent(s): a2c43fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ import ruclip
26
 
27
  # prepare models:
28
  #device = 'cuda'
29
- device = "cuda:0" if torch.cuda.is_available() else "CPU"
30
  #fp16=True
31
  #if device == 'CPU':
32
  # fp16 = False
 
26
 
27
  # prepare models:
28
  #device = 'cuda'
29
+ device = "cuda" if torch.cuda.is_available() else "cpu"
30
  #fp16=True
31
  #if device == 'CPU':
32
  # fp16 = False