michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix for one error with BSRGAN, though it fails to work anyway #1109

Browse source
This commit is contained in:
AUTOMATIC 2022-09-30 18:55:46 +03:00
parent 43c87ef0fc
commit ac21d308cd
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/bsrgan_model.py
View file

@ -67,9 +67,8 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler):
else: else:
filename = path filename = path
if not os.path.exists(filename) or filename is None: if not os.path.exists(filename) or filename is None:
print("Unable to load %s from %s" % (self.model_dir, filename)) print(f"BSRGAN: Unable to load model from {filename}", file=sys.stderr)
return None return None
print("Loading %s from %s" % (self.model_dir, filename))
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=2) # define network
model.load_state_dict(torch.load(filename), strict=True) model.load_state_dict(torch.load(filename), strict=True)
model.eval() model.eval()