jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

close llm on interrupt (#577)

Browse Source
This commit is contained in:
Bruce MacDonald 2023-09-22 19:41:52 +01:00 committed by GitHub
parent 9df31c3518
commit 5d71bda478
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/routes.go
View File

@ -570,6 +570,9 @@ func Serve(ln net.Listener, allowOrigins []string) error {
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM) signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
go func() { go func() {
<-signals <-signals
if loaded.llm != nil {
loaded.llm.Close()
}
os.RemoveAll(workDir) os.RemoveAll(workDir)
os.Exit(0) os.Exit(0)
}() }()