c9eace8371
Signed-off-by: iwzbi <iwzbi@zju.edu.cn> Signed-off-by: iwzbi <wzbi@zju.edu.cn> Co-authored-by: iwzbi <iwzbi@zju.edu.cn> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
16 lines
375 B
Python
16 lines
375 B
Python
"""Stable diffusion3 model components."""
|
|
|
|
from vllm_omni.diffusion.models.sd3.pipeline_sd3 import (
|
|
StableDiffusion3Pipeline,
|
|
get_sd3_image_post_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.sd3.sd3_transformer import (
|
|
SD3Transformer2DModel,
|
|
)
|
|
|
|
__all__ = [
|
|
"StableDiffusion3Pipeline",
|
|
"SD3Transformer2DModel",
|
|
"get_sd3_image_post_process_func",
|
|
]
|