[Bugfix][Diffusion] Fuse distilled LoRA weights before HSDP sharding (#6948)

Signed-off-by: samithuang <285365963@qq.com>
Signed-off-by: Samit <285365963@qq.com>
This commit is contained in:
Samit authored and GitHub committed 2026-09-06 01:29:59 +08:00
1 parent ae70479619
commit 44d3ae100a
9 files changed
+311 -1

No files matched your search

@@ -61,6 +61,9 @@ def test_video_pipeline_rejects_non_video_final_outputs(stage_configs):
"WanDMDPipeline",
"LingBotWorldCausalDMDPipeline",
"LongCatVideoAvatarPipeline",
"SanaVideoPipeline",
"SanaImageToVideoPipeline",
"SanaWmPipeline",
],
)
def test_registered_video_aliases_declare_video_output(model_class_name):