valhalla commited on
Commit
9b1c362
·
1 Parent(s): fd25d71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ if os.path.exists(result_path):
26
 
27
 
28
  device = "cpu"
29
- model = Dalle.from_pretrained("minDALL-E/1.3B") # This will automatically download the pretrained model.
30
  model.to(device=device)
31
 
32
  model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)
 
26
 
27
 
28
  device = "cpu"
29
+ model = Dalle.from_pretrained(result_path) # This will automatically download the pretrained model.
30
  model.to(device=device)
31
 
32
  model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)