jgitsolutions commited on
Commit
946ca97
·
verified ·
1 Parent(s): 3ee9494

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -502,7 +502,9 @@ class HatsStrategy(UpscalerStrategy):
502
  elif (psutil.cpu_count(logical=False) or 0) < 4 and Config.DEVICE == 'cpu':
503
  pass
504
  else:
505
- self.model = torch.compile(self.model)
 
 
506
  self.compiled = True
507
  except Exception:
508
  self.compiled = True
 
502
  elif (psutil.cpu_count(logical=False) or 0) < 4 and Config.DEVICE == 'cpu':
503
  pass
504
  else:
505
+ # HAT architecture also triggers "UntypedStorage" weakref errors with inductor on CPU
506
+ logger.info("ℹ Skipping torch.compile for HAT-S (incompatible architecture).")
507
+ # self.model = torch.compile(self.model)
508
  self.compiled = True
509
  except Exception:
510
  self.compiled = True