Skip to content

S
Stable Diffusion Webui
Commit 19eb1467 authored by d8ahazard's avatar d8ahazard
Browse files
Options

Fix BSRGAN variable not found.

parent 99aa132d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
modules/bsrgan_model.py
View file @ 19eb1467
...@@ -67,9 +67,8 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler): ...@@ -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("Unable to load %s from %s" % (self.model_path, filename))
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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment