cosmicdream commited on
Commit
0c8df61
1 Parent(s): b0ff8d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ import ruclip
29
  device = "cuda" if torch.cuda.is_available() else "cpu"
30
  print('Using device:', device)
31
 
32
- dalle = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
33
  tokenizer = get_tokenizer()
34
  vae = get_vae(dwt=True).to(device)
35
 
 
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)
33
  tokenizer = get_tokenizer()
34
  vae = get_vae(dwt=True).to(device)
35