Yueqian LinandHongsheng Liu
0cb7ec62da
Signed-off-by: linyueqian <linyueqian@outlook.com> Signed-off-by: Yueqian Lin <70319226+linyueqian@users.noreply.github.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
19 lines
522 B
Python
19 lines
522 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
"""Stable Audio Open model support for vLLM-Omni."""
|
|
|
|
from vllm_omni.diffusion.models.stable_audio.pipeline_stable_audio import (
|
|
StableAudioPipeline,
|
|
get_stable_audio_post_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.stable_audio.stable_audio_transformer import (
|
|
StableAudioDiTModel,
|
|
)
|
|
|
|
__all__ = [
|
|
"StableAudioDiTModel",
|
|
"StableAudioPipeline",
|
|
"get_stable_audio_post_process_func",
|
|
]
|