print

backend/lcm_text_to_image.py

@@ -243,7 +243,7 @@ class LCMTextToImage:
         newH = int(h * newW / w)
         lcm_diffusion_setting.image_height=newH
         lcm_diffusion_setting.init_image = lcm_diffusion_setting.init_image.resize((newW, newH)) 
-        print("修改图像尺寸了",newW,newH)
+        print("修改图像尺寸了",lcm_diffusion_setting.image_height,lcm_diffusion_setting.image_width)
         img_to_img_inference_steps = lcm_diffusion_setting.inference_steps
         check_step_value = int(
             lcm_diffusion_setting.inference_steps * lcm_diffusion_setting.strength
@@ -270,12 +270,12 @@ class LCMTextToImage:
             if reshape and not self.is_openvino_init:
                 print("Reshape and compile，调整尺寸")
                 
-                # self.pipeline.reshape(
-                #     batch_size=-1,
-                #     height=newH,
-                #     width=newW,
-                #     num_images_per_prompt=lcm_diffusion_setting.number_of_images,
-                # )
+                self.pipeline.reshape(
+                    batch_size=-1,
+                    height=newH,
+                    width=newW,
+                    num_images_per_prompt=lcm_diffusion_setting.number_of_images,
+                )
                 self.pipeline.compile()
 
             if self.is_openvino_init: