1movmedia/pyworker
3
0 Fork 0
Code Pull Requests Activity

Merge pull request #3 from Nader-gator/main

Browse Source 
add cuda error handling
This commit is contained in:
Nader Arbabian authored and GitHub committed 2025-04-09 12:17:11 -07:00
2 parents f15c7a3540 099bb5fa04
commit 28977c8bf9
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
workers/tgi/server.py
+1 -1
View File
@@ -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(