Yueqian LinandSy03
c111f3a2d6
Signed-off-by: Yueqian Lin <linyueqian@outlook.com> Co-authored-by: Sy03 <1370724210@qq.com>
17 lines
604 B
Python
17 lines
604 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
"""AuK diffusion-stage components: the flow transformer, the codec, the pipeline."""
|
|
|
|
from vllm_omni.diffusion.models.auk.auk_transformer import AuKTransformer, dit_state_dict, sample_latents
|
|
from vllm_omni.diffusion.models.auk.auk_vae import AuKVAE
|
|
from vllm_omni.diffusion.models.auk.pipeline_auk import AuKPipeline, get_auk_post_process_func
|
|
|
|
__all__ = [
|
|
"AuKPipeline",
|
|
"AuKTransformer",
|
|
"AuKVAE",
|
|
"dit_state_dict",
|
|
"get_auk_post_process_func",
|
|
"sample_latents",
|
|
]
|