change size

backend/lcm_text_to_image.py

@@ -238,11 +238,11 @@ class LCMTextToImage:
         reshape: bool = False,
     ) -> Any:
         guidance_scale = lcm_diffusion_setting.guidance_scale
-        w, h = lcm_diffusion_setting.init_image.size
-        newW = lcm_diffusion_setting.image_width
-        newH = int(h * newW / w)
-        lcm_diffusion_setting.image_height=newH
-        lcm_diffusion_setting.init_image = lcm_diffusion_setting.init_image.resize((newW, newH)) 
+        # w, h = lcm_diffusion_setting.init_image.size
+        # newW = lcm_diffusion_setting.image_width
+        # 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("修改图像尺寸了",lcm_diffusion_setting.image_height,lcm_diffusion_setting.image_width)
         img_to_img_inference_steps = lcm_diffusion_setting.inference_steps
         check_step_value = int(
@@ -272,8 +272,8 @@ class LCMTextToImage:
                 
                 self.pipeline.reshape(
                     batch_size=-1,
-                    height=newH,
-                    width=newW,
+                    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()