image shape

backend/lcm_text_to_image.py

@@ -272,8 +272,8 @@ class LCMTextToImage:
                 
                 self.pipeline.reshape(
                     batch_size=-1,
-                    height=lcm_diffusion_setting.image_height,
-                    width=lcm_diffusion_setting.image_width,
+                    height=256,
+                    width=256,
                     num_images_per_prompt=lcm_diffusion_setting.number_of_images,
                 )
                 self.pipeline.compile()
@@ -323,7 +323,6 @@ class LCMTextToImage:
                     negative_prompt=lcm_diffusion_setting.negative_prompt,
                     num_inference_steps=img_to_img_inference_steps * 3,
                     guidance_scale=guidance_scale,
-                    output_size=(256,256),
                     num_images_per_prompt=lcm_diffusion_setting.number_of_images,
                 ).images