Upload app.py
Browse files
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 |
-
|
|
|
|
|
|
|
| 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
|