Didan DengandHongsheng Liu
23cf48d7bb
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
22 lines
644 B
Python
22 lines
644 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Qwen Image diffusion model components."""
|
|
|
|
from vllm_omni.diffusion.models.qwen_image.cfg_parallel import (
|
|
QwenImageCFGParallelMixin,
|
|
)
|
|
from vllm_omni.diffusion.models.qwen_image.pipeline_qwen_image import (
|
|
QwenImagePipeline,
|
|
get_qwen_image_post_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.qwen_image.qwen_image_transformer import (
|
|
QwenImageTransformer2DModel,
|
|
)
|
|
|
|
__all__ = [
|
|
"QwenImageCFGParallelMixin",
|
|
"QwenImagePipeline",
|
|
"QwenImageTransformer2DModel",
|
|
"get_qwen_image_post_process_func",
|
|
]
|