From fbc4a807ead48b2281aa65fd99cde220dbcf1398 Mon Sep 17 00:00:00 2001
From: sczhou <shangchenzhou@gmail.com>
Date: Tue, 9 Aug 2022 13:34:10 +0800
Subject: [PATCH] update inference.

---
 inference_codeformer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/inference_codeformer.py b/inference_codeformer.py
index 65ae10e0..0de7b48f 100644
--- 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