Specify mega weights

app.py

@@ -46,13 +46,13 @@ model = MinDalle(
     models_root='./pretrained',
     dtype=float32,
     device='cpu',
-    is_mega=False,
+    is_mega=True,
     is_reusable=True
 )
 
 
 def gen_image(prompt):
-    torch.set_grad_enabled(False)
+    torch.set_grad_enabled(False)  # Hugging Space faces seems to run out of memory if grads are not disabled
     print(f'RUNNING gen_image with pronpt: {prompt}')
     images = model.generate_images(
         text=prompt,