Add tqdm `dynamic_ncols=True` (#3242)

2024-03-06 14:41:42 -08:00 · 2024-03-06 14:41:42 -08:00 · 4cb3b924cd
parent a33ce60c66
commit 4cb3b924cd
1 changed files with 3 additions and 1 deletions
--- a/vllm/entrypoints/llm.py
+++ b/vllm/entrypoints/llm.py
@ -191,7 +191,9 @@ class LLM:
        # Initialize tqdm.
        if use_tqdm:
            num_requests = self.llm_engine.get_num_unfinished_requests()
-            pbar = tqdm(total=num_requests, desc="Processed prompts")
+            pbar = tqdm(total=num_requests,
+                        desc="Processed prompts",
+                        dynamic_ncols=True)
        # Run the engine.
        outputs: List[RequestOutput] = []
        while self.llm_engine.has_unfinished_requests():