ollamarunner: Suppress stack trace during memory allocation

Allocation failures can be a normal part of new memory estimates, so
we shouldn't print a stack trace in this case.
This commit is contained in:
Jesse Gross
2025-09-11 13:48:51 -07:00
committed by Jesse Gross
parent 61fb912ca4
commit 26214125e8

View File

@@ -18,7 +18,6 @@ import (
"reflect"
"regexp"
"runtime"
"runtime/debug"
"strconv"
"strings"
"sync"
@@ -1101,9 +1100,13 @@ func (s *Server) allocModel(
// Convert memory allocation panics to errors
defer func() {
if r := recover(); r != nil {
debug.PrintStack()
if err, ok := r.(error); ok {
panicErr = err
var noMem ml.ErrNoMem
if errors.As(err, &noMem) {
panicErr = noMem
} else {
panic(r)
}
} else {
panic(r)
}