Merge pull request #14474 from akx/realesrgan-is-esrgan

Correct RealESRGAN expected architecture type to ESRGAN
This commit is contained in:
AUTOMATIC1111
2023-12-30 22:40:11 +03:00
committed by GitHub

View File

@@ -40,7 +40,7 @@ class UpscalerRealESRGAN(Upscaler):
info.local_data_path, info.local_data_path,
device=self.device, device=self.device,
half=(not cmd_opts.no_half and not cmd_opts.upcast_sampling), 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( return upscale_with_model(
mod, mod,