anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Model] Add native MAGI-2 Preview diffusion support (#5918)

Browse Source 
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
This commit is contained in:
Hongsheng Liu authored and GitHub committed 2026-09-05 15:32:30 +08:00
1 parent 13c617a140
commit d4096e01fe
78 files changed
+9896 -459
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/registry.py
+6
View File
@@ -296,6 +296,11 @@ _DIFFUSION_MODELS = {
"pipeline_sana_video_i2v",
"SanaImageToVideoPipeline",
),
"Magi2Pipeline": (
"magi2",
"pipeline_magi2",
"Magi2Pipeline",
),
"OmniVoicePipeline": (
"omnivoice",
"pipeline_omnivoice",
@@ -588,6 +593,7 @@ _DIFFUSION_POST_PROCESS_FUNCS = {
"LingBotVideoPipeline": "get_lingbot_video_post_process_func",
"SanaVideoPipeline": "get_sana_video_post_process_func",
"SanaImageToVideoPipeline": "get_sana_video_i2v_post_process_func",
"Magi2Pipeline": "get_magi2_post_process_func",
"OmniVoicePipeline": "get_omnivoice_post_process_func",
"SenseNovaU1Pipeline": "get_sensenova_u1_post_process_func",
"Cosmos3OmniDiffusersPipeline": "get_cosmos3_post_process_func",