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

This commit is contained in:
Jeffrey Morgan
2024-07-03 20:04:30 -04:00
committed by GitHub
parent 0d16eb310e
commit 4d71c559b2

View File

@@ -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) {