Просмотр исходного кода

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

Jeffrey Morgan 10 месяцев назад
Родитель
Сommit
4d71c559b2
1 измененных файлов с 1 добавлено и 1 удалено
  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) {