diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-30 20:09:18 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-30 20:09:18 +0000 |
| commit | f253790b40a71f2d88951c26224e0c80b6baea52 (patch) | |
| tree | 786b67f0cde4c15b3a0112e02495a51978ea9d26 /modules/bsrgan_model.py | |
| parent | 3f64e23f81618395428da0a132bf3a6cfd5c42c9 (diff) | |
| parent | a9a0320d2688a292fe834986c9abe141ff29bf7c (diff) | |
| download | stable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.tar.gz stable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.tar.bz2 stable-diffusion-webui-gfx803-f253790b40a71f2d88951c26224e0c80b6baea52.zip | |
Merge pull request #1371 from d8ahazard/master
Fix LDSR, BSRGAN, recursive SD checkpoint Loading
Diffstat (limited to 'modules/bsrgan_model.py')
| -rw-r--r-- | modules/bsrgan_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/bsrgan_model.py b/modules/bsrgan_model.py index 47346f31..e62c6657 100644 --- a/modules/bsrgan_model.py +++ b/modules/bsrgan_model.py @@ -69,7 +69,7 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler): if not os.path.exists(filename) or filename is None: print(f"BSRGAN: Unable to load model from {filename}", file=sys.stderr) return None - model = RRDBNet(in_nc=3, out_nc=3, nf=64, nb=23, gc=32, sf=2) # define network + model = RRDBNet(in_nc=3, out_nc=3, nf=64, nb=23, gc=32, sf=4) # define network model.load_state_dict(torch.load(filename), strict=True) model.eval() for k, v in model.named_parameters(): |