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 |
-
self.pipeline.reshape(
|
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:
|