Add logging import and set logger level to WARNING
This commit is contained in:
1 parent
586ccbff1b
commit
6cb3acdd64
1 file changed
+3
@@ -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
|
||||
|
||||
Reference in new issue
Block a user