15 lines
375 B
Python
15 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",
|
||
|
|
]
|