ANHDYandwangmengdi
28467dc19a
Signed-off-by: wangmengdi <160612767@qq.com> Co-authored-by: wangmengdi <wangmengdi@cmss.chinamobile.com>
18 lines
546 B
Python
18 lines
546 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""HiDream Image diffusion model components."""
|
|
|
|
from vllm_omni.diffusion.models.hidream_image.hidream_image_transformer import (
|
|
HiDreamImageTransformer2DModel,
|
|
)
|
|
from vllm_omni.diffusion.models.hidream_image.pipeline_hidream_image import (
|
|
HiDreamImagePipeline,
|
|
get_hidream_image_post_process_func,
|
|
)
|
|
|
|
__all__ = [
|
|
"HiDreamImageTransformer2DModel",
|
|
"HiDreamImagePipeline",
|
|
"get_hidream_image_post_process_func",
|
|
]
|