Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix esrgan

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-10 18:16:59 -04:00
parent 56e041c3b6
commit 19d92dae52
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/esrgan_model.py
+1 -1
View File
@@ -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]