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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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