25 lines
827 B
Python
25 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",
|
||
|
|
]
|