tzhouam
617c6ca55f
feat: enhance model loading with prefetching and retry logic
...
- Introduced `from_pretrained_with_prefetch` to handle racy cache scenarios by re-prefetching and retrying on failures.
- Updated various model pipelines to utilize the new prefetching mechanism, ensuring robust loading of model components.
- Prefetch logic added to multiple models, including HunyuanVideo, LongCatImage, and QwenImage, to mitigate issues with incomplete cache states.
2026-06-02 09:01:47 +00:00
8297570c2b
[AutoRound] Support WAN2.2 W4A16 quantization model ( #3353 )
...
Signed-off-by: lvliang-intel <liang1.lv@intel.com >
Signed-off-by: hyh_hh <huyinghong1@huawei.com >
Co-authored-by: hxhhhlalala <hyh_hh@163.com >
Co-authored-by: hyh_hh <huyinghong1@huawei.com >
2026-05-21 17:54:41 +08:00
hxhhhlalala and hyh_hh
9dd36e3afd
[NPU][Quant] Add W4A4 MXFP4 online & MXFP4 dual-scale online/offline quantization support for Wan2.2 T2V / I2V inference on Ascend NPU ( #3578 )
...
Signed-off-by: hyh_hh <huyinghong1@huawei.com >
Co-authored-by: hyh_hh <huyinghong1@huawei.com >
2026-05-20 06:21:29 +00:00
Rustam Khadipash
89f8819525
[Feature] Add support for Pipeline Parallel and integrate it into Wan 2.2 ( #2322 )
...
Signed-off-by: Rustam Khadipash <16683750+hadipash@users.noreply.github.com >
2026-05-19 09:42:17 +08:00
hxhhhlalala and hyh_hh
a715abd447
[NPU][Quant] Add W8A8 MXFP8 online/offline quantization support for Wan2.2 T2V / I2V / TI2V inference on Ascend NPU ( #3140 )
...
Signed-off-by: hyh_hh <huyinghong1@huawei.com >
Co-authored-by: hyh_hh <huyinghong1@huawei.com >
2026-05-13 17:14:42 +08:00
56ca7dd612
support online FP8 quantization for FA on NPU #2236 ( #2640 )
...
Signed-off-by: lyj-jjj <liuyingjun5@huawei.com >
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
Co-authored-by: Cursor <cursoragent@cursor.com >
Co-authored-by: gcanlin <canlinguosdu@gmail.com >
2026-05-13 11:46:00 +08:00
Canlin Guo and Hongsheng Liu
11c4c7f0ff
[Diffusion][Attention] Support per-role attention backend via CLI ( #2681 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-05-10 11:59:05 +08:00
Canlin Guo
4a24a517ab
[BugFix] Probe __dict__ instead of hasattr when patching WanRMS_norm ( #3400 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
2026-05-07 11:54:16 +08:00
Chendi.Xue and Hongsheng Liu
4e41b7bc23
Enable Wan2.2-S2V modeling to vLLM-omni ( #2751 )
...
Signed-off-by: Chendi Xue <chendi.xue@intel.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-05-06 08:03:48 +08:00
Zhou Taichang and Yuanheng Zhao
7f2b1e9386
[Rebase] Rebase to vllm 0.20.0 ( #3232 )
...
Signed-off-by: tzhouam <tzhouam@connect.ust.hk >
Signed-off-by: Zhou Taichang <tzhouam@connect.ust.hk >
Signed-off-by: Yuanheng Zhao <jonathan.zhaoyh@gmail.com >
Co-authored-by: Yuanheng Zhao <54058983+yuanheng-zhao@users.noreply.github.com >
2026-04-30 09:56:59 +12:00
lyj-jjj
19980bb37c
[Perf][Wan2.2] Add fused RMSNorm replace WanRMS_norm on npu ( #3067 )
...
Signed-off-by: lyj-jjj <liuyingjun5@huawei.com >
2026-04-23 18:25:10 +00:00
Hui. and WeiQing Chen
95a07f7732
[PERF]use mindiesd fused rope and rope cache ( #2571 )
...
Signed-off-by: Hui <1779066624@qq.com >
Signed-off-by: Hui. <62495465+Hu1Lcode@users.noreply.github.com >
Co-authored-by: WeiQing Chen <40507679+david6666666@users.noreply.github.com >
2026-04-23 17:47:43 +08:00
bjf-frz
5f029ee4d5
[Enhancement]remove duplicate video preprocess in Wan2.2 pipeline ( #2963 )
...
Signed-off-by: bjf-frz <frz123db@gmail.com >
2026-04-21 20:33:09 +08:00
Ayush Agarwal
dc8a9e2f17
[Model] feat: FastGen DMD2-distilled Wan 2.1 pipelines (T2V, I2V) ( #2749 )
...
Signed-off-by: ayushag <ayushag@nvidia.com >
2026-04-20 21:21:43 +08:00
WeiQing Chen
c2859e9cd7
[Revert] drop Wan2.2 prompt-length enforcement from #2847 ( #2877 )
...
Signed-off-by: david6666666 <530634352@qq.com >
2026-04-20 19:19:38 +08:00
fan2956
78f237e603
[Perf] Optimize Wan2.2 device free on image preprocess ( #2852 )
...
Signed-off-by: fan2956 <zhoufan53@huawei.com >
2026-04-20 02:36:22 +00:00
bjf-frz
18ac679763
[Refactor] refactor wan2.2 diffuse && add ut ( #2672 )
...
Signed-off-by: bjf-frz <frz123db@gmail.com >
2026-04-17 20:13:49 +08:00
WeiQing Chen
3079e943de
[Bugfix] enforce max_sequence_length for Qwen-Image and Wan2.2 series before encoding ( #2847 )
...
Signed-off-by: david6666666 <530634352@qq.com >
2026-04-17 10:40:36 +08:00
fan2956 and gcanlin
f1cb4ebe4c
[PERF] Wan2.2 support rmsnorm fused op ( #2583 )
...
Signed-off-by: fan2956 <zhoufan53@huawei.com >
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
Co-authored-by: gcanlin <canlinguosdu@gmail.com >
2026-04-16 03:21:34 -07:00
fan2956 and Canlin Guo
2dff2d7c74
[PERF] Wan2.2 support adalayernorm fused op ( #2585 )
...
Signed-off-by: fan2956 <zhoufan53@huawei.com >
Co-authored-by: Canlin Guo <canlinguosdu@gmail.com >
2026-04-15 14:02:41 +08:00
WeiQing Chen
6c6551dff8
[Feature] feat: add video frame interpolation postprocess ( #2555 )
...
Signed-off-by: David Chen <530634352@qq.com >
2026-04-15 11:52:20 +08:00
Jiaqian Liu and Canlin Guo
8097747a5d
[Model] Adapt Wan2.2-I2V-A14B via LightX2V offline conversion path ( #2134 )
...
Signed-off-by: Celeste-jq <591998922@qq.com >
Co-authored-by: Canlin Guo <canlinguosdu@gmail.com >
2026-04-13 14:20:04 +08:00
Y. Fisher and Canlin Guo
ec082add35
[Perf] Wan2.2 I2V optimization: convert datatype from FP32 to BF16 in vae ( #2391 )
...
Signed-off-by: KexiongYu <yukexiong1@huawei.com >
Signed-off-by: Canlin Guo <961750412@qq.com >
Co-authored-by: Canlin Guo <961750412@qq.com >
2026-04-08 03:26:39 +00:00
R0CKSTAR
7a72f34ce4
Add supports_float64() to OmniPlatform and clean up MPS ( #2488 )
...
Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com >
2026-04-07 11:15:02 +08:00
Lancer and Didan Deng
328de586aa
[Feat] support for multi-block layerwise offloading, fix top-level parameters/buffers staying on CPU ( #1486 )
...
Signed-off-by: Lancer <maruixiang6688@gmail.com >
Signed-off-by: Lancer <402430575@qq.com >
Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
2026-04-06 11:49:35 +08:00
Canlin Guo
6dc61c9a20
[Perf] Skip Wan2.2 cross attn Ulysses SP ( #2459 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
2026-04-03 16:23:36 +08:00
458f402323
Add VACE support for WAN 2.1 conditional video generation ( #1885 )
...
Signed-off-by: Binh Tang <tangbinhna@gmail.com >
Signed-off-by: Binh Tang <binht@netflix.com >
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
Co-authored-by: Binh Tang <binht@netflix.com >
Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
2026-04-02 15:42:50 +08:00
Canlin Guo
e2892ef6c1
[Perf] Optimize Wan2.2 rotary embedding ( #2393 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
2026-04-02 14:28:44 +08:00
Lancer and gcanlin
4cfa9bf843
[Bugfix] fix offload and hsdp incompatibility ( #1888 )
...
Signed-off-by: Lancer <maruixiang6688@gmail.com >
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
Co-authored-by: gcanlin <canlinguosdu@gmail.com >
2026-03-24 14:50:57 +08:00
fde4808723
[Performance] diffusion timing ( #1757 )
...
Signed-off-by: dengyunyang <584797741@qq.com >
Signed-off-by: bjf-frz <frz123db@gmail.com >
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
Co-authored-by: bjf-frz <frz123db@gmail.com >
Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
2026-03-19 12:26:12 +08:00
Samit
155856ff30
[Perf] Reduce IPC overhead for single-stage diffusion serving for Wan2.2 ( #1715 )
...
Signed-off-by: samithuang <285365963@qq.com >
Signed-off-by: Samit <285365963@qq.com >
2026-03-10 00:03:51 +13:00
Canlin Guo
d5ddc190f9
[UX] Add progress bar for diffusion models ( #1652 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
2026-03-05 23:56:14 +08:00
675c35db97
[Refactor] lora: reuse load_weights packed mapping ( #991 )
...
Signed-off-by: dongbo910220 <1275604947@qq.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Jiangyun Zhu <riverclouds.zhu@qq.com >
2026-03-04 02:58:43 +00:00
dengyunyang
d275c0c993
[FEATURE] Tile/Patch parallelism refactor for easily support other models ( #1366 )
...
Signed-off-by: dengyunyang <584797741@qq.com >
2026-03-03 15:48:51 +08:00
Canlin Guo and Hongsheng Liu
f53152d340
[Feature] Support HSDP for diffusion models ( #1339 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-02-26 21:31:10 +08:00
Canlin Guo
9913c00144
[Feature] Support Wan2.2 output with irregular shapes ( #1279 )
...
Signed-off-by: gcanlin <canlinguosdu@gmail.com >
2026-02-26 17:03:43 +08:00
Samit and Flora Feng
9c00852d85
[Feature] Support Wan2.2 T2V and I2V Online Serving with OpenAI /v1/videos API ( #1073 )
...
Signed-off-by: samithuang <285365963@qq.com >
Signed-off-by: Samit <285365963@qq.com >
Signed-off-by: SamitHuang <285365963@qq.com >
Co-authored-by: Flora Feng <4florafeng@gmail.com >
2026-02-12 14:28:32 +08:00
XU Mingshi and Hongsheng Liu
11a5a914ca
[Bugfix] Fix wan2.2 ti2v ( #1221 )
...
Signed-off-by: mxuax <mxuax@connect.ust.hk >
Signed-off-by: XU Mingshi <91017482+mxuax@users.noreply.github.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-02-06 17:28:49 +08:00
weichen
c4933ec2aa
[Wan 2.2][Diffusion] Add TP Support ( #964 )
...
Signed-off-by: weichen <calvin_zhu0210@outlook.com >
2026-02-05 18:11:12 +08:00
Didan Deng and Hongsheng Liu
23cf48d7bb
[Perf]: CFG parallel abstraction ( #851 )
...
Signed-off-by: Didan Deng <33117903+wtomin@users.noreply.github.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-01-30 16:23:44 +00:00
2923d5a91e
[Feature] Support DiT Layerwise (Blockwise) CPU Offloading ( #858 )
...
Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com >
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com >
2026-01-30 17:35:09 +08:00
Fanli Lin and Samit
c7f89ef085
[Wan2.2] Optimize memory usage with conditional transformer loading ( #980 )
...
Signed-off-by: Lin, Fanli <fanli.lin@intel.com >
Signed-off-by: Samit <285365963@qq.com >
Co-authored-by: Samit <285365963@qq.com >
2026-01-29 22:02:24 +08:00
Zeyu Huang | 黃澤宇
d64bbde561
[Frontend][Model] Support batch request with refined OmniDiffusionReq… ( #797 )
...
Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com >
2026-01-28 16:07:42 +08:00
Jiangyun Zhu and Hongsheng Liu
1c8fcf436f
[Perf] replace torch's conv with vLLM's conv to fix torch 2.9 performance regression ( #982 )
...
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-01-28 10:41:33 +08:00
Yuanheng Zhao
2f5e32477c
[NFC] Remove redundant torch.no_grad in models and pipelines ( #854 )
...
Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com >
2026-01-27 12:36:54 +00:00
XU Mingshi and Hongsheng Liu
1fe8e71433
[Diffusion][Feature] Non-Intrusive Sequence Parallelism (SP) Support for Wan2.2 ( #966 )
...
Signed-off-by: mxuax <mxuax@connect.ust.hk >
Signed-off-by: XU Mingshi <91017482+mxuax@users.noreply.github.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-01-27 11:33:38 +08:00
5037af1033
[Feature] Diffusion LoRA Adapter Support (PEFT compatible) for vLLM alignment ( #758 )
...
Signed-off-by: AndyZhou952 <jzhoubc@connect.ust.hk >
Signed-off-by: Andy Zhou <jzhoubc@connect.ust.hk >
Signed-off-by: dongbo910220 <1275604947@qq.com >
Signed-off-by: Andy Zhou <46011930+AndyZhou952@users.noreply.github.com >
Signed-off-by: Samit <285365963@qq.com >
Co-authored-by: dongbo910220 <1275604947@qq.com >
Co-authored-by: Samit <285365963@qq.com >
2026-01-27 10:22:40 +08:00
Samit and Hongsheng Liu
978a290d6e
[Bugfix] Fix diffusion pipeline CFG (guidance_scale parsing failure bug) ( #922 )
...
Signed-off-by: samithuang <285365963@qq.com >
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-01-26 05:09:03 +00:00
Yueqian Lin and Hongsheng Liu
3fb6adc40a
[bugfix] use unipc scheduler for Wan 2.2 ( #804 )
...
Signed-off-by: linyueqian <linyueqian@outlook.com >
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com >
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com >
2026-01-16 12:31:25 +00:00
Yueqian Lin
aaea77a180
[bugfix] Fix Wan2.2 I2V warmup failure by adding support_image_input attribute ( #791 )
...
Signed-off-by: linyueqian <linyueqian@outlook.com >
2026-01-15 13:14:34 +00:00