From 4d71c559b21ec9207a328b824ce534bdbaf59f2d Mon Sep 17 00:00:00 2001 From: Jeffrey Morgan Date: Wed, 3 Jul 2024 20:04:30 -0400 Subject: [PATCH] fix error detection by limiting model loading error parsing (#5472) --- llm/status.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llm/status.go b/llm/status.go index 0f56b7f9..d9f36115 100644 --- 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) {