dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-28 22:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 12

feat(shutdown): allow force shutdown of backends (#3733)

Browse source 
We default to a soft kill, however, we might want to force killing
backends after a while to avoid hanging requests (which may hallucinate
indefinetly)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-10-05 10:41:35 +02:00 • committed by GitHub
parent 905473c739
commit e28e80857b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pkg/model/loader.go
View file

@ -157,6 +157,11 @@ func (ml *ModelLoader) ShutdownModel(modelName string) error {
} }
time.Sleep(dur) time.Sleep(dur)
retries++ retries++
if retries > 10 && os.Getenv("LOCALAI_FORCE_BACKEND_SHUTDOWN") == "true" {
log.Warn().Msgf("Model %s is still busy after %d retries. Forcing shutdown.", modelName, retries)
break
}
} }
return ml.deleteProcess(modelName) return ml.deleteProcess(modelName)