da03 commited on
Commit
497c0a8
·
1 Parent(s): 836e4c4
Files changed (1) hide show
  1. main.py +1 -1
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 = []