Browse Source

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

Jeffrey Morgan 10 months ago
parent
commit
4d71c559b2
1 changed files with 1 additions and 1 deletions
  1. 1 1
      llm/status.go

+ 1 - 1
llm/status.go

@@ -25,7 +25,7 @@ var errorPrefixes = []string{
 	"CUDA error",
 	"cudaMalloc failed",
 	"\"ERR\"",
-	"architecture",
+	"error loading model",
 }
 
 func (w *StatusWriter) Write(b []byte) (int, error) {