michaelj commited on
Commit
9469b65
·
1 Parent(s): b9718d3

change faster

Browse files
Files changed (1) hide show
  1. backend/lcm_text_to_image.py +5 -5
backend/lcm_text_to_image.py CHANGED
@@ -5,8 +5,8 @@ from backend.models.lcmdiffusion_setting import LCMDiffusionSetting
5
  import numpy as np
6
  from constants import DEVICE
7
  from backend.models.lcmdiffusion_setting import LCMLora
8
- from backend.utils_sd import register_normal_pipeline, register_faster_forward, register_parallel_pipeline, seed_everything # 1.import package
9
- print("我加载了")
10
  from backend.device import is_openvino_device
11
  from backend.openvino.pipelines import (
12
  get_ov_text_to_image_pipeline,
@@ -328,10 +328,10 @@ class LCMTextToImage:
328
  lcm_diffusion_setting.diffusion_task
329
  == DiffusionTask.image_to_image.value
330
  ):
331
- register_parallel_pipeline(self.pipeline) # 2. enable parallel. If memory is limited, replace it with `register_normal_pipeline(pipe)`
332
- register_faster_forward(self.pipeline.unet) # 3. encoder propagation
333
 
334
- result_images = self.pipeline.call(
335
  image=lcm_diffusion_setting.init_image,
336
  strength=lcm_diffusion_setting.strength,
337
  prompt=lcm_diffusion_setting.prompt,
 
5
  import numpy as np
6
  from constants import DEVICE
7
  from backend.models.lcmdiffusion_setting import LCMLora
8
+ # from backend.utils_sd import register_normal_pipeline, register_faster_forward, register_parallel_pipeline, seed_everything # 1.import package
9
+ # print("我加载了")
10
  from backend.device import is_openvino_device
11
  from backend.openvino.pipelines import (
12
  get_ov_text_to_image_pipeline,
 
328
  lcm_diffusion_setting.diffusion_task
329
  == DiffusionTask.image_to_image.value
330
  ):
331
+ # register_parallel_pipeline(self.pipeline) # 2. enable parallel. If memory is limited, replace it with `register_normal_pipeline(pipe)`
332
+ # register_faster_forward(self.pipeline.unet) # 3. encoder propagation
333
 
334
+ result_images = self.pipeline(
335
  image=lcm_diffusion_setting.init_image,
336
  strength=lcm_diffusion_setting.strength,
337
  prompt=lcm_diffusion_setting.prompt,