Skip to content

Commit fbc4a80

Browse files
committed
update inference.
1 parent 12cfcfc commit fbc4a80

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

inference_codeformer.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@
3838
if not torch.cuda.is_available(): # CPU
3939
import warnings
4040
warnings.warn('The unoptimized RealESRGAN is slow on CPU. We do not use it. '
41-
'If you really want to use it, please modify the corresponding codes.')
41+
'If you really want to use it, please modify the corresponding codes.',
42+
category=RuntimeWarning)
4243
bg_upsampler = None
4344
else:
4445
from basicsr.archs.rrdbnet_arch import RRDBNet

0 commit comments

Comments
 (0)