baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix infinite keep_alive (#2480)

Browse source
This commit is contained in:
bnorick 2024-02-13 15:40:32 -08:00 • committed by GitHub
parent 1d263449ff
commit caf2b13c10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
api/types.go
View file

@ -415,8 +415,7 @@ func (d *Duration) UnmarshalJSON(b []byte) (err error) {
switch t := v.(type) {
case float64:
if t < 0 {
t = math.MaxFloat64
d.Duration = time.Duration(t)
d.Duration = time.Duration(math.MaxInt64)
} else {
d.Duration = time.Duration(t * float64(time.Second))
}
@ -426,8 +425,7 @@ func (d *Duration) UnmarshalJSON(b []byte) (err error) {
return err
}
if d.Duration < 0 {
mf := math.MaxFloat64
d.Duration = time.Duration(mf)
d.Duration = time.Duration(math.MaxInt64)
}
}