catch more specific errors

2021-11-24 00:14:05 +08:00
parent 35ee6f781e
commit 7dd860a881
4 changed files with 4 additions and 4 deletions
--- a/realesrgan/utils.py
+++ b/realesrgan/utils.py
@@ -128,7 +128,7 @@ class RealESRGANer():
                try:
                    with torch.no_grad():
                        output_tile = self.model(input_tile)
-                except Exception as error:
+                except RuntimeError as error:
                    print('Error', error)
                print(f'\tTile {tile_idx}/{tiles_x * tiles_y}')