faster

backend/lcm_text_to_image.py

@@ -5,6 +5,8 @@ from backend.models.lcmdiffusion_setting import LCMDiffusionSetting
 import numpy as np
 from constants import DEVICE
 from backend.models.lcmdiffusion_setting import LCMLora
+from utils_sd import register_normal_pipeline, register_faster_forward, register_parallel_pipeline, seed_everything  # 1.import package
+
 from backend.device import is_openvino_device
 from backend.openvino.pipelines import (
     get_ov_text_to_image_pipeline,
@@ -326,7 +328,10 @@ class LCMTextToImage:
                 lcm_diffusion_setting.diffusion_task
                 == DiffusionTask.image_to_image.value
             ):
-                result_images = self.pipeline(
+                register_parallel_pipeline(self.pipeline) # 2. enable parallel. If memory is limited, replace it with  `register_normal_pipeline(pipe)`
+                register_faster_forward(self.pipeline.unet)  # 3. encoder propagation
+
+                result_images = self.pipeline.call(
                     image=lcm_diffusion_setting.init_image,
                     strength=lcm_diffusion_setting.strength,
                     prompt=lcm_diffusion_setting.prompt,