diff --git a/scripts/postprocessing_codeformer.py b/scripts/postprocessing_codeformer.py
index 53a0cc44c..9052ed4a1 100644
--- a/scripts/postprocessing_codeformer.py
+++ b/scripts/postprocessing_codeformer.py
@@ -29,6 +29,14 @@ class ScriptPostprocessingCodeFormer(scripts_postprocessing.ScriptPostprocessing
         res = Image.fromarray(restored_img)
 
         if codeformer_visibility < 1.0:
+            # Ensure consistent size
+            if pp.image.size != res.size:
+                res = res.resize(pp.image.size)
+                
+            # Ensure consistent mode
+            if pp.image.mode != res.mode:
+                res = res.convert(pp.image.mode)
+                
             res = Image.blend(pp.image, res, codeformer_visibility)
 
         pp.image = res