16 lines
604 B
Python
16 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",
|
||
|
|
]
|