From 3b066f8e29e929cd38d30f0203012f01faa0e6f8 Mon Sep 17 00:00:00 2001 From: Bryce Date: Mon, 15 May 2023 08:23:25 -0700 Subject: [PATCH] fix: don't hide error messages during upscale --- imaginairy/vendored/realesrgan.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/imaginairy/vendored/realesrgan.py b/imaginairy/vendored/realesrgan.py index 29c5d10..936f8ba 100644 --- a/imaginairy/vendored/realesrgan.py +++ b/imaginairy/vendored/realesrgan.py @@ -172,11 +172,8 @@ class RealESRGANer: ] # upscale tile - try: - with torch.no_grad(): - output_tile = self.model(input_tile) - except RuntimeError as error: - print("Error", error) + with torch.no_grad(): + output_tile = self.model(input_tile) # output tile area on total image output_start_x = input_start_x * self.scale