change width and height
Browse files
backend/lcm_text_to_image.py
CHANGED
|
@@ -272,8 +272,8 @@ class LCMTextToImage:
|
|
| 272 |
|
| 273 |
self.pipeline.reshape(
|
| 274 |
batch_size=-1,
|
| 275 |
-
height=
|
| 276 |
-
width=
|
| 277 |
num_images_per_prompt=lcm_diffusion_setting.number_of_images,
|
| 278 |
)
|
| 279 |
self.pipeline.compile()
|
|
|
|
| 272 |
|
| 273 |
self.pipeline.reshape(
|
| 274 |
batch_size=-1,
|
| 275 |
+
height=newH,
|
| 276 |
+
width=newW,
|
| 277 |
num_images_per_prompt=lcm_diffusion_setting.number_of_images,
|
| 278 |
)
|
| 279 |
self.pipeline.compile()
|