From 097c918cb99656a654c7bb17ca4455f55c05d0b0 Mon Sep 17 00:00:00 2001 From: Yannic Kilcher Date: Fri, 10 Feb 2023 23:30:42 +0100 Subject: [PATCH] bugfix --- inference/worker/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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...")