Unverified Commit 4d71c559 authored by Jeffrey Morgan's avatar Jeffrey Morgan Committed by GitHub
Browse files

fix error detection by limiting model loading error parsing (#5472)

parent 0d16eb31
...@@ -25,7 +25,7 @@ var errorPrefixes = []string{ ...@@ -25,7 +25,7 @@ var errorPrefixes = []string{
"CUDA error", "CUDA error",
"cudaMalloc failed", "cudaMalloc failed",
"\"ERR\"", "\"ERR\"",
"architecture", "error loading model",
} }
func (w *StatusWriter) Write(b []byte) (int, error) { func (w *StatusWriter) Write(b []byte) (int, error) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment