change shape
Browse files
backend/lcm_text_to_image.py
CHANGED
@@ -269,12 +269,12 @@ class LCMTextToImage:
|
|
269 |
print("Using OpenVINO")
|
270 |
if reshape and not self.is_openvino_init:
|
271 |
print("Reshape and compile,调整尺寸")
|
272 |
-
|
273 |
-
|
274 |
-
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
self.pipeline.compile()
|
279 |
|
280 |
if self.is_openvino_init:
|
|
|
269 |
print("Using OpenVINO")
|
270 |
if reshape and not self.is_openvino_init:
|
271 |
print("Reshape and compile,调整尺寸")
|
272 |
+
self.pipeline.reshape(
|
273 |
+
batch_size=-1,
|
274 |
+
height=lcm_diffusion_setting.image_height,
|
275 |
+
width=lcm_diffusion_setting.image_width,
|
276 |
+
num_images_per_prompt=lcm_diffusion_setting.number_of_images,
|
277 |
+
)
|
278 |
self.pipeline.compile()
|
279 |
|
280 |
if self.is_openvino_init:
|