michaelj commited on
Commit
acb1f51
·
1 Parent(s): cf62e43
Files changed (1) hide show
  1. backend/lcm_text_to_image.py +6 -3
backend/lcm_text_to_image.py CHANGED
@@ -238,6 +238,11 @@ class LCMTextToImage:
238
  reshape: bool = False,
239
  ) -> Any:
240
  guidance_scale = lcm_diffusion_setting.guidance_scale
 
 
 
 
 
241
  img_to_img_inference_steps = lcm_diffusion_setting.inference_steps
242
  check_step_value = int(
243
  lcm_diffusion_setting.inference_steps * lcm_diffusion_setting.strength
@@ -263,9 +268,7 @@ class LCMTextToImage:
263
  print("Using OpenVINO")
264
  if reshape and not self.is_openvino_init:
265
  print("Reshape and compile,调整尺寸")
266
- w, h = lcm_diffusion_setting.init_image.size
267
- newW = lcm_diffusion_setting.image_width
268
- newH = int(h * newW / w)
269
  self.pipeline.reshape(
270
  batch_size=-1,
271
  height=newH,
 
238
  reshape: bool = False,
239
  ) -> Any:
240
  guidance_scale = lcm_diffusion_setting.guidance_scale
241
+ w, h = lcm_diffusion_setting.init_image.size
242
+ newW = lcm_diffusion_setting.image_width
243
+ newH = int(h * newW / w)
244
+ lcm_diffusion_setting.init_image = lcm_diffusion_setting.init_image.resize((newW, newH))
245
+ print("修改图像尺寸了",newW,newH)
246
  img_to_img_inference_steps = lcm_diffusion_setting.inference_steps
247
  check_step_value = int(
248
  lcm_diffusion_setting.inference_steps * lcm_diffusion_setting.strength
 
268
  print("Using OpenVINO")
269
  if reshape and not self.is_openvino_init:
270
  print("Reshape and compile,调整尺寸")
271
+
 
 
272
  self.pipeline.reshape(
273
  batch_size=-1,
274
  height=newH,