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