update inference.

sczhou · Aug 9, 2022 · fbc4a80 · fbc4a80
1 parent 12cfcfc
commit fbc4a80
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/inference_codeformer.py b/inference_codeformer.py
@@ -38,7 +38,8 @@
         if not torch.cuda.is_available():  # CPU
             import warnings
             warnings.warn('The unoptimized RealESRGAN is slow on CPU. We do not use it. '
-                          'If you really want to use it, please modify the corresponding codes.')
+                          'If you really want to use it, please modify the corresponding codes.',
+                          category=RuntimeWarning)
             bg_upsampler = None
         else:
             from basicsr.archs.rrdbnet_arch import RRDBNet