diff --git a/workers/openai/worker.py b/workers/openai/worker.py index 46ce751..50d0b9a 100644 --- a/workers/openai/worker.py +++ b/workers/openai/worker.py @@ -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