Update vllm/model_executor/model_loader/weight_utils.py

bbartels · DarkLight1337 · web-flow · commit d3505ba831a9 · 2025-07-22T15:59:34.000+01:00
Co-authored-by: Cyrus Leung &lt;cyrus.tl.leung@gmail.com&gt;
diff --git a/vllm/model_executor/model_loader/weight_utils.py b/vllm/model_executor/model_loader/weight_utils.py
@@ -492,7 +492,7 @@ def runai_safetensors_weights_iterator(
             total=total_tensors,
             desc="Loading safetensors using Runai Model Streamer",
             bar_format=_BAR_FORMAT,
-            disable=not use_tqdm_on_load,
+            disable=not enable_tqdm(use_tqdm_on_load),
         )
 
         yield from tensor_iter

Original file line number	Diff line number	Diff line change
`@@ -492,7 +492,7 @@ def runai_safetensors_weights_iterator(`
`492`	`492`	`total=total_tensors,`
`493`	`493`	`desc="Loading safetensors using Runai Model Streamer",`
`494`	`494`	`bar_format=_BAR_FORMAT,`
`495`		`- disable=not use_tqdm_on_load,`
	`495`	`+ disable=not enable_tqdm(use_tqdm_on_load),`
`496`	`496`	`)`
`497`	`497`
`498`	`498`	`yield from tensor_iter`