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

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: