michaelj commited on
Commit
aa03471
·
1 Parent(s): a8e0053

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
@@ -270,12 +270,12 @@ class LCMTextToImage:
270
  if reshape and not self.is_openvino_init:
271
  print("Reshape and compile,调整尺寸")
272
 
273
- self.pipeline.reshape(
274
- batch_size=-1,
275
- height=newH,
276
- width=newW,
277
- num_images_per_prompt=lcm_diffusion_setting.number_of_images,
278
- )
279
  self.pipeline.compile()
280
 
281
  if self.is_openvino_init:
 
270
  if reshape and not self.is_openvino_init:
271
  print("Reshape and compile,调整尺寸")
272
 
273
+ # self.pipeline.reshape(
274
+ # batch_size=-1,
275
+ # height=newH,
276
+ # width=newW,
277
+ # num_images_per_prompt=lcm_diffusion_setting.number_of_images,
278
+ # )
279
  self.pipeline.compile()
280
 
281
  if self.is_openvino_init: