From 19d92dae526cf2e2da142fb62a8d17844044b45e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 10 Sep 2023 18:16:59 -0400 Subject: [PATCH] fix esrgan --- modules/esrgan_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py index a7656ad58..7ed5c33ec 100644 --- a/modules/esrgan_model.py +++ b/modules/esrgan_model.py @@ -201,7 +201,7 @@ def upscale_without_tiling(model, img): img = img.unsqueeze(0).to(devices.device_esrgan) with devices.inference_context(): output = model(img) - output = output.squeeze().float().cpu().clamp_(0, 1).numpy() + output = output.squeeze().float().cpu().clamp_(0, 1).detach().numpy() output = 255. * np.moveaxis(output, 0, 2) output = output.astype(np.uint8) output = output[:, :, ::-1]