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",
"cudaMalloc failed",
"\"ERR\"",
"architecture",
"error loading model",
}
func (w *StatusWriter) Write(b []byte) (int, error) {