Add logging import and set logger level to WARNING

2026-05-21 19:50:21 +00:00
parent 586ccbff1b
commit 6cb3acdd64
1 changed files with 3 additions and 0 deletions
@@ -1,9 +1,12 @@
 import random
 import os
 import re
+import logging

 from vastai import Worker, WorkerConfig, HandlerConfig, LogActionConfig, BenchmarkConfig

+logging.getLogger().setLevel(logging.WARNING)  # Only show warnings and errors
+
 # Ollama model configuration
 MODEL_SERVER_URL           = 'http://127.0.0.1'
 MODEL_SERVER_PORT          = 11434