[Refactor]Base class for dit pipelines with unified parameter declaration[1/N] (#4225)

Signed-off-by: princepride <wangzhipeng628@gmail.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
authored and GitHub committed 2026-06-18 17:14:18 +08:00
1 parent dff8cda844
commit f75c437a60
26 files changed
+1026 -1586

No files matched your search

-4
View File
@@ -27,8 +27,6 @@ from vllm_omni.diffusion.data import (
from vllm_omni.diffusion.executor.abstract import DiffusionExecutor
from vllm_omni.diffusion.io_support import (
get_dummy_run_num_frames,
get_extra_body_params,
get_extra_output_params,
image_color_format,
supports_audio_output,
supports_multimodal_input,
@@ -61,8 +59,6 @@ __all__ = [
"_RpcTask",
"_move_tensor_tree_to_cpu",
"get_dummy_run_num_frames",
"get_extra_body_params",
"get_extra_output_params",
"image_color_format",
"supports_audio_output",
"supports_multimodal_input",