diff --git a/vllm_omni/diffusion/utils/hf_utils.py b/vllm_omni/diffusion/utils/hf_utils.py index 61c7f1982..2fd3b1a23 100644 --- a/vllm_omni/diffusion/utils/hf_utils.py +++ b/vllm_omni/diffusion/utils/hf_utils.py @@ -1,12 +1,13 @@ from functools import lru_cache -from diffusers.pipelines.pipeline_utils import DiffusionPipeline from vllm.logger import init_logger logger = init_logger(__name__) def load_diffusers_config(model_name) -> dict: + from diffusers.pipelines.pipeline_utils import DiffusionPipeline + config = DiffusionPipeline.load_config(model_name) return config