Spaces:
Paused
Paused
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:
|