Correct RealESRGAN expected architecture type to ESRGAN

2023-12-30 21:12:32 +02:00 · 2023-12-30 21:12:32 +02:00 · 393a5b82ba
parent 16848f950b
commit 393a5b82ba
1 changed files with 1 additions and 1 deletions
--- a/modules/realesrgan_model.py
+++ b/modules/realesrgan_model.py
@ -40,7 +40,7 @@ class UpscalerRealESRGAN(Upscaler):
            info.local_data_path,
            device=self.device,
            half=(not cmd_opts.no_half and not cmd_opts.upcast_sampling),
-            expected_architecture="RealESRGAN",
+            expected_architecture="ESRGAN",  # "RealESRGAN" isn't a specific thing for Spandrel
        )
        return upscale_with_model(
            mod,