Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
497c0a8
1
Parent(s):
836e4c4
main.py
CHANGED
|
@@ -146,7 +146,7 @@ model = initialize_model("standard_challenging_context32_nocond_all.yaml", "yunt
|
|
| 146 |
|
| 147 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 148 |
model = model.to(device)
|
| 149 |
-
model = torch.compile(model)
|
| 150 |
|
| 151 |
def load_initial_images(width, height):
|
| 152 |
initial_images = []
|
|
|
|
| 146 |
|
| 147 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 148 |
model = model.to(device)
|
| 149 |
+
#model = torch.compile(model)
|
| 150 |
|
| 151 |
def load_initial_images(width, height):
|
| 152 |
initial_images = []
|