Spaces:

cosmicdream
/

Image_Variations

Runtime error

cosmicdream commited on Jul 17, 2022

Commit

48226ad

1 Parent(s): 2f265c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,9 +26,9 @@ import ruclip
 # prepare models:
 #device = 'cuda'
-device1 = "cuda:0" if torch.cuda.is_available() else "CPU"
 fp16=True
-if device1 == 'CPU':
     fp16 = False
 model = get_rudalle_model('Malevich', pretrained=True, fp16=fp16, device=device)

 # prepare models:
 #device = 'cuda'
+device = "cuda:0" if torch.cuda.is_available() else "CPU"
 fp16=True
+if device == 'CPU':
     fp16 = False
 model = get_rudalle_model('Malevich', pretrained=True, fp16=fp16, device=device)