change width and height
Browse files
backend/lcm_text_to_image.py
CHANGED
@@ -323,6 +323,8 @@ class LCMTextToImage:
|
|
323 |
negative_prompt=lcm_diffusion_setting.negative_prompt,
|
324 |
num_inference_steps=img_to_img_inference_steps * 3,
|
325 |
guidance_scale=guidance_scale,
|
|
|
|
|
326 |
num_images_per_prompt=lcm_diffusion_setting.number_of_images,
|
327 |
).images
|
328 |
|
|
|
323 |
negative_prompt=lcm_diffusion_setting.negative_prompt,
|
324 |
num_inference_steps=img_to_img_inference_steps * 3,
|
325 |
guidance_scale=guidance_scale,
|
326 |
+
width=256,
|
327 |
+
height=256,
|
328 |
num_images_per_prompt=lcm_diffusion_setting.number_of_images,
|
329 |
).images
|
330 |
|