diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-12-31 19:52:32 +0200 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2024-01-02 10:44:38 +0200 |
| commit | 2cacbc124c49f45da5b66b79d9b0a3ab943472eb (patch) | |
| tree | 27ae05c022710aaa60e7791ca6bde3e7f60b7511 /modules/realesrgan_model.py | |
| parent | 51f1cca8524d3ffa8930b32a571d239c60d65725 (diff) | |
load_spandrel_model: make `half` `prefer_half`
As discussed with the Spandrel folks, it's good to heed Spandrel's
"supports half precision" flag to avoid e.g. black blotches and what-not.
Diffstat (limited to 'modules/realesrgan_model.py')
| -rw-r--r-- | modules/realesrgan_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/realesrgan_model.py b/modules/realesrgan_model.py index 4d35b695..ff9d8ac0 100644 --- a/modules/realesrgan_model.py +++ b/modules/realesrgan_model.py @@ -39,7 +39,7 @@ class UpscalerRealESRGAN(Upscaler): model_descriptor = modelloader.load_spandrel_model(
info.local_data_path,
device=self.device,
- half=(not cmd_opts.no_half and not cmd_opts.upcast_sampling),
+ prefer_half=(not cmd_opts.no_half and not cmd_opts.upcast_sampling),
expected_architecture="ESRGAN", # "RealESRGAN" isn't a specific thing for Spandrel
)
return upscale_with_model(
|