Emaad commited on
Commit
adef0eb
1 Parent(s): 774daca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class model:
46
 
47
  # Instantiate model from config and move to device
48
  self.model = instantiate_from_config(config.model).to(device)
49
- self.model = torch.compile(self.model,mode='max-autotune')
50
 
51
  os.chdir(base_path)
52
 
 
46
 
47
  # Instantiate model from config and move to device
48
  self.model = instantiate_from_config(config.model).to(device)
49
+ self.model = torch.compile(self.model)
50
 
51
  os.chdir(base_path)
52