diffusion: fix wan-2.2-TI2V and support sp (#12926)

2025-11-10 14:37:57 +08:00
parent 90401cf7d2
commit e123648b36
12 changed files with 296 additions and 89 deletions
--- a/python/sglang/multimodal_gen/runtime/utils/hf_diffusers_utils.py
+++ b/python/sglang/multimodal_gen/runtime/utils/hf_diffusers_utils.py
@@ -370,7 +370,7 @@ def maybe_download_model(
        logger.info(
            "Downloading model snapshot from HF Hub for %s...", model_name_or_path
        )
-        with get_lock(model_name_or_path):
+        with get_lock(model_name_or_path).acquire(poll_interval=2):
            local_path = snapshot_download(
                repo_id=model_name_or_path,
                ignore_patterns=["*.onnx", "*.msgpack"],