JMalott commited on
Commit
890d938
1 Parent(s): 848f278

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -105,10 +105,10 @@ def generate2(prompt,crazy,k):
105
 
106
  mm = MinDalle(
107
  models_root='./pretrained',
108
- dtype=torch.float16,
109
  device='cpu',
110
  is_mega=False,
111
- is_reusable=True
112
  )
113
 
114
 
 
105
 
106
  mm = MinDalle(
107
  models_root='./pretrained',
108
+ dtype=torch.float32,
109
  device='cpu',
110
  is_mega=False,
111
+ is_reusable=False
112
  )
113
 
114