change shape

backend/lcm_text_to_image.py

@@ -93,12 +93,12 @@ class LCMTextToImage:
         lcm_lora: LCMLora = lcm_diffusion_setting.lcm_lora
         ov_model_id = lcm_diffusion_setting.openvino_lcm_model_id
 
-        if lcm_diffusion_setting.diffusion_task == DiffusionTask.image_to_image.value:
-            lcm_diffusion_setting.init_image = resize_pil_image(
-                lcm_diffusion_setting.init_image,
-                lcm_diffusion_setting.image_width,
-                lcm_diffusion_setting.image_height,
-            )
+        # if lcm_diffusion_setting.diffusion_task == DiffusionTask.image_to_image.value:
+        #     lcm_diffusion_setting.init_image = resize_pil_image(
+        #         lcm_diffusion_setting.init_image,
+        #         lcm_diffusion_setting.image_width,
+        #         lcm_diffusion_setting.image_height,
+        #     )
 
         if (
             self.pipeline is None
@@ -269,11 +269,10 @@ class LCMTextToImage:
             print("Using OpenVINO")
             if reshape and not self.is_openvino_init:
                 print("Reshape and compile，调整尺寸")
-                
                 self.pipeline.reshape(
                     batch_size=-1,
-                    height=256,
-                    width=256,
+                    height=lcm_diffusion_setting.image_height,
+                    width=lcm_diffusion_setting.image_width,
                     num_images_per_prompt=lcm_diffusion_setting.number_of_images,
                 )
                 self.pipeline.compile()