diff --git a/inference/worker/__main__.py b/inference/worker/__main__.py index f5ce9794..74cf4d1d 100644 --- a/inference/worker/__main__.py +++ b/inference/worker/__main__.py @@ -16,7 +16,7 @@ def main(): def on_open(ws: websocket.WebSocket): logger.info("Connected to backend, sending config...") - worker_config = inference.WorkerConfig(model_name=settings.model_name) + worker_config = inference.WorkerConfig(model_name=settings.model_id) ws.send(worker_config.json()) logger.info("Config sent, waiting for work...")