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

4d71c559 · Jeffrey Morgan · GitHub · 0d16eb31 · 4d71c559
Unverified Commit 4d71c559 authored Jul 03, 2024 by Jeffrey Morgan Committed by GitHub Jul 03, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

llm/status.go llm/status.go +1 -1

No files found.
--- a/llm/status.go
+++ b/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) {