diff --git a/basicsr/models/base_model.py b/basicsr/models/base_model.py
index bf1f90ac..b4021175 100644
--- a/basicsr/models/base_model.py
+++ b/basicsr/models/base_model.py
@@ -5,7 +5,7 @@
 from copy import deepcopy
 from torch.nn.parallel import DataParallel, DistributedDataParallel
 
-from basicsr.models import lr_scheduler as lr_scheduler
+#from basicsr.models import lr_scheduler as lr_scheduler
 from basicsr.utils.dist_util import master_only
 
 logger = logging.getLogger('basicsr')
diff --git a/inference_codeformer.py b/inference_codeformer.py
index 1a38cc95..171b37a5 100644
--- a/inference_codeformer.py
+++ b/inference_codeformer.py
@@ -259,6 +259,10 @@ def set_realesrgan():
         img_list = sorted(glob.glob(os.path.join(result_root, 'final_results', '*.[jp][pn]g')))
         for img_path in img_list:
             img = cv2.imread(img_path)
+            height, width = img.shape[:2]
+            wb = width % 2
+            hb = height % 2
+            img = cv2.copyMakeBorder(img, 0, hb, wb, 0, cv2.BORDER_REFLECT)
             video_frames.append(img)
         # write images to video
         height, width = video_frames[0].shape[:2]