Correct RealESRGAN expected architecture type to ESRGAN

This commit is contained in:
Aarni Koskela 2023-12-30 21:12:32 +02:00
parent 16848f950b
commit 393a5b82ba

View File

@ -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,