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 |
|