dsinghviandHongsheng Liu
4586efe651
Signed-off-by: Divyansh Singhvi <divyanshsinghvi@gmail.com> Signed-off-by: dsinghvi <divyanshsinghvi@gmail.com> Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
18 lines
516 B
Python
18 lines
516 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""Ovis Image 7B diffusion model components."""
|
|
|
|
from vllm_omni.diffusion.models.ovis_image.ovis_image_transformer import (
|
|
OvisImageTransformer2DModel,
|
|
)
|
|
from vllm_omni.diffusion.models.ovis_image.pipeline_ovis_image import (
|
|
OvisImagePipeline,
|
|
get_ovis_image_post_process_func,
|
|
)
|
|
|
|
__all__ = [
|
|
"OvisImagePipeline",
|
|
"OvisImageTransformer2DModel",
|
|
"get_ovis_image_post_process_func",
|
|
]
|