Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -105,10 +105,10 @@ def generate2(prompt,crazy,k):
|
|
| 105 |
|
| 106 |
mm = MinDalle(
|
| 107 |
models_root='./pretrained',
|
| 108 |
-
dtype=torch.
|
| 109 |
device='cpu',
|
| 110 |
is_mega=False,
|
| 111 |
-
is_reusable=
|
| 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 |
|