diff --git a/workers/tgi/server.py b/workers/tgi/server.py index 84faaeb..2b881ad 100644 --- a/workers/tgi/server.py +++ b/workers/tgi/server.py @@ -15,7 +15,7 @@ MODEL_SERVER_URL = "http://0.0.0.0:5001" # This is the last log line that gets emitted once comfyui+extensions have been fully loaded MODEL_SERVER_START_LOG_MSG = '"message":"Connected","target":"text_generation_router"' -MODEL_SERVER_ERROR_LOG_MSGS = ["Error: WebserverFailed", "Error: DownloadError"] +MODEL_SERVER_ERROR_LOG_MSGS = ["Error: WebserverFailed", "Error: DownloadError", "Error: ShardCannotStart"] logging.basicConfig(