diff options
Diffstat (limited to 'modules/hat_model.py')
| -rw-r--r-- | modules/hat_model.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/hat_model.py b/modules/hat_model.py index 553e1941..7f2abb41 100644 --- a/modules/hat_model.py +++ b/modules/hat_model.py @@ -39,4 +39,5 @@ class UpscalerHAT(Upscaler): return modelloader.load_spandrel_model(
path,
device=devices.device_esrgan, # TODO: should probably be device_hat
+ expected_architecture='HAT',
)
|