Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Aug 8, 2022

Commit

ab6e890

1 Parent(s): 9278b39

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -107,7 +107,7 @@ def generate2(prompt,crazy,k):
         models_root='./pretrained',
         dtype=torch.float32,
         device='cpu',
-        is_mega=False,
         is_reusable=True
     )

         models_root='./pretrained',
         dtype=torch.float32,
         device='cpu',
+        is_mega=True,
         is_reusable=True
     )