BruceLoveDecimaland汪志鹏
adf6d0c1ea
Signed-off-by: BruceLoveDecimal <liuqihao970610@gmail.com> Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com> Signed-off-by: liuqihao <liuqihao970610@gmail.com> Co-authored-by: 汪志鹏 <wangzhipeng628@gmail.com>
26 lines
827 B
Python
26 lines
827 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""SANA-WM diffusion model integration."""
|
|
|
|
from vllm_omni.diffusion.models.sana_wm.config import SanaWmConfig
|
|
from vllm_omni.diffusion.models.sana_wm.pipeline_sana_wm import (
|
|
SANA_WM_MODEL_ID,
|
|
SANA_WM_OUTPUT_HEIGHT,
|
|
SANA_WM_OUTPUT_WIDTH,
|
|
SanaWmPipeline,
|
|
get_sana_wm_pre_process_func,
|
|
)
|
|
from vllm_omni.diffusion.models.sana_wm.request import normalize_sana_wm_payload
|
|
from vllm_omni.diffusion.models.sana_wm.sana_wm_transformer import SanaWmTransformer3DModel
|
|
|
|
__all__ = [
|
|
"SANA_WM_MODEL_ID",
|
|
"SANA_WM_OUTPUT_HEIGHT",
|
|
"SANA_WM_OUTPUT_WIDTH",
|
|
"SanaWmConfig",
|
|
"SanaWmPipeline",
|
|
"SanaWmTransformer3DModel",
|
|
"get_sana_wm_pre_process_func",
|
|
"normalize_sana_wm_payload",
|
|
]
|