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 = []
|