Merge pull request #1129 from modelscope/ascend

Support Ascend NPU
2026-03-22 08:40:47 +00:00 · 2025-12-15 19:13:40 +08:00
parent 47545e1aab 7c6905a432
commit bfaaf12bf4
13 changed files with 244 additions and 9 deletions
--- a/diffsynth/pipelines/wan_video.py
+++ b/diffsynth/pipelines/wan_video.py
@@ -126,7 +126,7 @@ class WanVideoPipeline(BasePipeline):
        pipe = WanVideoPipeline(device=device, torch_dtype=torch_dtype)
        if use_usp:
            from ..utils.xfuser import initialize_usp
-            initialize_usp()
+            initialize_usp(device)
        model_pool = pipe.download_and_load_models(model_configs, vram_limit)
        
        # Fetch models