Skip to content

Commit 4236d14

Browse files
committed
fix(pu): fix vllm logging level
1 parent 7638cdd commit 4236d14

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

zoo/jericho/priorzero/priorzero_entry_sync.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -457,5 +457,10 @@ def main():
457457
# Now set tokenizers env var
458458
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
459459

460+
# [LOG-FIX] Reduce vLLM logging verbosity to avoid flooding logs with progress bars
461+
# This suppresses the frequent "Processed prompts: XX%" messages from vLLM
462+
os.environ['VLLM_LOGGING_LEVEL'] = 'WARNING' # Suppress INFO level logs from vLLM
463+
os.environ['VLLM_CONFIGURE_LOGGING'] = '0' # Disable vLLM's logging configuration
464+
460465
# Finally, run main
461466
main()

zoo/jericho/priorzero/utils/vllm_engine.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,7 @@ def create_vllm_engines(
177177
num_gpus=0.2 if use_hybrid_engine else 1,
178178
enable_sleep_mode=vllm_enable_sleep,
179179
agent_func_path=agent_func_path,
180+
disable_log_stats=True, # [LOG-FIX] Disable verbose progress bars
180181
**additional_kwargs,
181182
)
182183
)

0 commit comments

Comments
 (0)