Spaces:

cosmicdream
/

Image_Variations

Runtime error

cosmicdream commited on Jul 18, 2022

Commit

0c8df61

1 Parent(s): b0ff8d1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ import ruclip
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print('Using device:', device)
-dalle = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=True).to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 print('Using device:', device)
+model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=True).to(device)