Commit ac21d308 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

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

parent 43c87ef0
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -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(f"BSRGAN: Unable to load model from {filename}", file=sys.stderr)
            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()