Emaad commited on
Commit
815b096
1 Parent(s): 8d2fbeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ class model:
41
 
42
  # Instantiate model from config and move to device
43
  self.model = instantiate_from_config(config.model).to(device)
44
- self.model = torch.compile(self.model,mode='reduce-overhead')
45
 
46
  os.chdir(base_path)
47
 
 
41
 
42
  # Instantiate model from config and move to device
43
  self.model = instantiate_from_config(config.model).to(device)
44
+ self.model = torch.compile(self.model,mode='max-autotune')
45
 
46
  os.chdir(base_path)
47