Add logging import and set logger level to WARNING
This commit is contained in:
@@ -1,9 +1,12 @@
|
|||||||
import random
|
import random
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
|
import logging
|
||||||
|
|
||||||
from vastai import Worker, WorkerConfig, HandlerConfig, LogActionConfig, BenchmarkConfig
|
from vastai import Worker, WorkerConfig, HandlerConfig, LogActionConfig, BenchmarkConfig
|
||||||
|
|
||||||
|
logging.getLogger().setLevel(logging.WARNING) # Only show warnings and errors
|
||||||
|
|
||||||
# Ollama model configuration
|
# Ollama model configuration
|
||||||
MODEL_SERVER_URL = 'http://127.0.0.1'
|
MODEL_SERVER_URL = 'http://127.0.0.1'
|
||||||
MODEL_SERVER_PORT = 11434
|
MODEL_SERVER_PORT = 11434
|
||||||
|
|||||||
Reference in New Issue
Block a user