diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-30 09:05:00 -0500 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-30 09:05:00 -0500 |
| commit | 19eb1467f1acbb45d3b416ad7887ba9ab8a94e75 (patch) | |
| tree | d677e01fc85d0e5b3c9758eb0652724f6403eb57 | |
| parent | 99aa132df7045077a420918d276fcca877fdc9e3 (diff) | |
Fix BSRGAN variable not found.
| -rw-r--r-- | modules/bsrgan_model.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/modules/bsrgan_model.py b/modules/bsrgan_model.py index 77141545..339f402b 100644 --- a/modules/bsrgan_model.py +++ b/modules/bsrgan_model.py @@ -67,9 +67,8 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler): else: filename = path 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 - 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.load_state_dict(torch.load(filename), strict=True) model.eval() |