Commit af95be6
Fix Wan and Motif video pipeline fast test failures (#14269)
* Fix Wan and Motif video pipeline fast test failures
Description
This PR fixes several failing fast tests for Motif Video and Wan pipelines.
Changes
Updated test_motif_video.py to use a Motif-specific fp16 save/load tolerance.
T5Gemma2Encoder rebuilds non-persistent RoPE buffers after save/load, which introduces small fp16 numerical drift. The test now keeps coverage while allowing this expected drift.
Updated transformer_wan.py to keep Wan RoPE modules in fp32 during mixed precision handling.
This avoids fp16 precision drift in Wan save/load paths.
Updated memory.py group offloading coverage to include transformer_2 and skip None optional components.
This fixes Wan 2.2 memory/offloading tests where optional secondary transformers may be absent.
Updated stale expected slices in test_wan_image_to_video.py for:
TestWanImageToVideoPipeline::test_inference
TestWanFLFToVideoPipeline::test_inference
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
* revert change in test_wan_image_to_video.py
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
---------
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>1 parent 4d89a88 commit af95be6
3 files changed
Lines changed: 8 additions & 1 deletion
File tree
- src/diffusers/models/transformers
- tests/pipelines
- motif_video
- testing_utils
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
546 | 546 | | |
547 | 547 | | |
548 | 548 | | |
549 | | - | |
| 549 | + | |
550 | 550 | | |
551 | 551 | | |
552 | 552 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
141 | 141 | | |
142 | 142 | | |
143 | 143 | | |
| 144 | + | |
| 145 | + | |
| 146 | + | |
| 147 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
288 | 288 | | |
289 | 289 | | |
290 | 290 | | |
| 291 | + | |
291 | 292 | | |
292 | 293 | | |
293 | 294 | | |
294 | 295 | | |
295 | 296 | | |
296 | 297 | | |
| 298 | + | |
| 299 | + | |
297 | 300 | | |
298 | 301 | | |
299 | 302 | | |
| |||
0 commit comments