fix error detection by limiting model loading error parsing (#5472)
This commit is contained in:
parent
0d16eb310e
commit
4d71c559b2
1 changed files with 1 additions and 1 deletions
|
@ -25,7 +25,7 @@ var errorPrefixes = []string{
|
|||
"CUDA error",
|
||||
"cudaMalloc failed",
|
||||
"\"ERR\"",
|
||||
"architecture",
|
||||
"error loading model",
|
||||
}
|
||||
|
||||
func (w *StatusWriter) Write(b []byte) (int, error) {
|
||||
|
|
Loading…
Reference in a new issue