change shape
Browse files
backend/lcm_text_to_image.py
CHANGED
|
@@ -270,12 +270,12 @@ class LCMTextToImage:
|
|
| 270 |
if reshape and not self.is_openvino_init:
|
| 271 |
print("Reshape and compile,调整尺寸")
|
| 272 |
|
| 273 |
-
self.pipeline.reshape(
|
| 274 |
-
|
| 275 |
-
|
| 276 |
-
|
| 277 |
-
|
| 278 |
-
)
|
| 279 |
self.pipeline.compile()
|
| 280 |
|
| 281 |
if self.is_openvino_init:
|
|
|
|
| 270 |
if reshape and not self.is_openvino_init:
|
| 271 |
print("Reshape and 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()
|
| 280 |
|
| 281 |
if self.is_openvino_init:
|