change size

backend/lcm_text_to_image.py

@@ -262,11 +262,14 @@ class LCMTextToImage:
         if is_openvino_pipe:
             print("Using OpenVINO")
             if reshape and not self.is_openvino_init:
-                print("Reshape and compile")
+                print("Reshape and compile，调整尺寸")
+                w, h = lcm_diffusion_setting.init_image.size
+                newW = lcm_diffusion_setting.image_width
+                newH = int(h * newW / w)
                 self.pipeline.reshape(
                     batch_size=-1,
-                    height=lcm_diffusion_setting.image_height,
-                    width=lcm_diffusion_setting.image_width,
+                    height=newH,
+                    width=newW,
                     num_images_per_prompt=lcm_diffusion_setting.number_of_images,
                 )
                 self.pipeline.compile()