diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-29 07:38:51 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-06-13 09:44:25 +0000 |
| commit | bf67a5dcf44c3dbd88d1913478d4e02477915f33 (patch) | |
| tree | e4c1705c31bb76eef77781ef6df0628e2716f929 /extensions-builtin/LDSR | |
| parent | e3a973a68df3cfe13039dae33d19cf2c02a741e0 (diff) | |
| download | stable-diffusion-webui-gfx803-bf67a5dcf44c3dbd88d1913478d4e02477915f33.tar.gz stable-diffusion-webui-gfx803-bf67a5dcf44c3dbd88d1913478d4e02477915f33.tar.bz2 stable-diffusion-webui-gfx803-bf67a5dcf44c3dbd88d1913478d4e02477915f33.zip | |
Upscaler.load_model: don't return None, just use exceptions
Diffstat (limited to 'extensions-builtin/LDSR')
| -rw-r--r-- | extensions-builtin/LDSR/scripts/ldsr_model.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/extensions-builtin/LDSR/scripts/ldsr_model.py b/extensions-builtin/LDSR/scripts/ldsr_model.py index bf9b6de2..bd78dece 100644 --- a/extensions-builtin/LDSR/scripts/ldsr_model.py +++ b/extensions-builtin/LDSR/scripts/ldsr_model.py @@ -46,16 +46,13 @@ class UpscalerLDSR(Upscaler): yaml = local_yaml_path or load_file_from_url(self.yaml_url, model_dir=self.model_download_path, file_name="project.yaml") - try: - return LDSR(model, yaml) - except Exception: - errors.report("Error importing LDSR", exc_info=True) - return None + return LDSR(model, yaml) def do_upscale(self, img, path): - ldsr = self.load_model(path) - if ldsr is None: - print("NO LDSR!") + try: + ldsr = self.load_model(path) + except Exception: + errors.report(f"Failed loading LDSR model {path}", exc_info=True) return img ddim_steps = shared.opts.ldsr_steps return ldsr.super_resolution(img, ddim_steps, self.scale) |