michaelj commited on
Commit
29dda49
·
1 Parent(s): 8e08b7b

change shape

Browse files
Files changed (1) hide show
  1. backend/lcm_text_to_image.py +6 -6
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
- 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:
 
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: