Update inference_i2mv_sdxl.py

inference_i2mv_sdxl.py

@@ -187,12 +187,24 @@ def run_pipeline(
     control_images = ((plucker_embeds + 1.0) / 2.0).clamp(0, 1)
 
     # Prepare image
+    # reference_image = Image.open(image) if isinstance(image, str) else image
+    # if remove_bg_fn is not None:
+    #     reference_image = remove_bg_fn(reference_image)
+    #     reference_image = preprocess_image(reference_image, height, width)
+    # elif reference_image.mode == "RGBA":
+    #     reference_image = preprocess_image(reference_image, height, width)
     reference_image = Image.open(image) if isinstance(image, str) else image
+    logging.info(f"Initial reference_image mode: {reference_image.mode}")
+    
     if remove_bg_fn is not None:
+        logging.info("Using remove_bg_fn")
         reference_image = remove_bg_fn(reference_image)
         reference_image = preprocess_image(reference_image, height, width)
     elif reference_image.mode == "RGBA":
+        logging.info("Image is RGBA, preprocessing directly")
         reference_image = preprocess_image(reference_image, height, width)
+    
+    logging.info(f"Final reference_image mode: {reference_image.mode}")
 
     pipe_kwargs = {}
     if seed != -1 and isinstance(seed, int):