Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Aug 9, 2022

Commit

3095ffa

1 Parent(s): d546093

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -106,7 +106,7 @@ def generate2(prompt,crazy,k):
     mm = MinDalle(
         models_root='./pretrained',
         dtype=torch.float32,
-        device='cuda',
         is_mega=False,
         is_reusable=False
     )
@@ -119,7 +119,7 @@ def generate2(prompt,crazy,k):
         is_seamless=False,
         temperature=crazy,
         top_k=k,#2128,
-        supercondition_factor=32,
         is_verbose=False
     )

     mm = MinDalle(
         models_root='./pretrained',
         dtype=torch.float32,
+        device='cpu',
         is_mega=False,
         is_reusable=False
     )
         is_seamless=False,
         temperature=crazy,
         top_k=k,#2128,
+        supercondition_factor=16,
         is_verbose=False
     )