This commit is contained in:
Yannic Kilcher
2023-02-10 23:30:42 +01:00
parent 91e5b11315
commit 097c918cb9
+1 -1
View File
@@ -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...")