Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 6, 2022

Commit

54d3e7c

•

1 Parent(s): a59a4d4

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -110,7 +110,7 @@ def generate2(prompt,crazy,k):
     mm = MinDalle(
         models_root='./pretrained',
-        dtype=torch.float16,
         device='cpu',
         is_mega=False,
         is_reusable=False

     mm = MinDalle(
         models_root='./pretrained',
+        dtype=torch.float32,
         device='cpu',
         is_mega=False,
         is_reusable=False