9bf8f8ce79
Signed-off-by: Lancer <maruxiiang6688@gmail.com> Signed-off-by: Lancer <maruixiang6688@gmail.com> Signed-off-by: Lancer <402430575@qq.com> Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com> Signed-off-by: Gao Han <hgaoaf@connect.ust.hk> Co-authored-by: Lancer <maruxiiang6688@gmail.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com> Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
26 lines
727 B
Python
26 lines
727 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""FLUX diffusion model components."""
|
|
|
|
from vllm_omni.diffusion.models.flux.flux_transformer import (
|
|
FluxKontextTransformer2DModel,
|
|
FluxTransformer2DModel,
|
|
)
|
|
from vllm_omni.diffusion.models.flux.pipeline_flux import (
|
|
FluxPipeline,
|
|
get_flux_post_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.flux.pipeline_flux_kontext import (
|
|
FluxKontextPipeline,
|
|
get_flux_kontext_post_process_func,
|
|
)
|
|
|
|
__all__ = [
|
|
"FluxPipeline",
|
|
"FluxKontextPipeline",
|
|
"FluxTransformer2DModel",
|
|
"FluxKontextTransformer2DModel",
|
|
"get_flux_post_process_func",
|
|
"get_flux_kontext_post_process_func",
|
|
]
|