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:
parent
61fb912ca4
commit
26214125e8
|
|
@ -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)
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue