fix: missing tqdm import in weight_loader

2026-05-16 06:31:14 +00:00
parent 8496ac99bc
commit dab31b0961
1 changed files with 2 additions and 1 deletions
--- a/vllm/patches/deepseek_v4.py
+++ b/vllm/patches/deepseek_v4.py
@@ -355,7 +355,8 @@ class DeepseekV4MegaMoEExperts(nn.Module):
    ) -> bool:
        # Progress bar for k8s/docker liveness during GPU upload
        if DeepseekV4MegaMoEExperts._weight_load_count == 0:
-            DeepseekV4MegaMoEExperts._weight_load_tqdm = tqdm(
+            from tqdm import tqdm as _tqdm
+            DeepseekV4MegaMoEExperts._weight_load_tqdm = _tqdm(
                total=self.num_local_experts * 20,  # ~20 tensors per expert
                desc="  Loading Native NVFP4 Expert Weights",
                unit="tensor",