fix colorfix for the input size was different

modules/colorfix.py

@@ -21,6 +21,8 @@ def adain_color_fix(target: Image, source: Image):
     return result_image
 
 def wavelet_color_fix(target: Image, source: Image):
+    source = source.resize(target.size, resample=Image.Resampling.LANCZOS)
+
     # Convert images to tensors
     to_tensor = ToTensor()
     target_tensor = to_tensor(target).unsqueeze(0)

nodes.py

@@ -48,18 +48,12 @@ def INPUT_TYPES(s):
     CATEGORY = "image"
 
     def fix_color(self, image, color_map_image, color_fix):
-        print(f"[StableSR] fix_color")
-        try:
-            color_fix_func = (
-                wavelet_color_fix if color_fix == "Wavelet" else adain_color_fix
-            )
-            result_image = color_fix_func(
-                tensor2pil(image), tensor2pil(color_map_image)
-            )
-            refined_image = pil2tensor(result_image)
-            return (refined_image,)
-        except Exception as e:
-            print(f"[StableSR] Error fix_color: {e}")
+        color_fix_func = (
+            wavelet_color_fix if color_fix == "Wavelet" else adain_color_fix
+        )
+        result_image = color_fix_func(tensor2pil(image), tensor2pil(color_map_image))
+        refined_image = pil2tensor(result_image)
+        return (refined_image,)
 
 
 original_sample = comfy.sample.sample