Hugging Face's logo

Spaces:
cosmicdream
/
Image_Variations
Runtime error

App Files Community
2
cosmicdream commited on
Commit
5a71d15
•
1 Parent(s): 1dc335e

Update app.py

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