15 lines
588 B
Python
15 lines
588 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""
|
|
ErnieImage diffusion model for vLLM-Omni.
|
|
|
|
This module implements ERNIE-Image text-to-image generation with:
|
|
- ErnieImageTransformer2DModel: Custom DiT transformer
|
|
- ErnieImagePipeline: Full generation pipeline
|
|
"""
|
|
|
|
from vllm_omni.diffusion.models.ernie_image.ernie_image_transformer import ErnieImageTransformer2DModel
|
|
from vllm_omni.diffusion.models.ernie_image.pipeline_ernie_image import ErnieImagePipeline
|
|
|
|
__all__ = ["ErnieImageTransformer2DModel", "ErnieImagePipeline"]
|