Skip to content

Commit 76fba02

Browse files
authored
fix: do not keep track model if not existing (#8603)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
1 parent dadc715 commit 76fba02

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

pkg/model/process.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,15 @@ import (
1717

1818
var forceBackendShutdown bool = os.Getenv("LOCALAI_FORCE_BACKEND_SHUTDOWN") == "true"
1919

20+
var (
21+
modelNotFoundErr = errors.New("model not found")
22+
)
23+
2024
func (ml *ModelLoader) deleteProcess(s string) error {
2125
model, ok := ml.models[s]
2226
if !ok {
2327
xlog.Debug("Model not found", "model", s)
24-
return fmt.Errorf("model %s not found", s)
28+
return modelNotFoundErr
2529
}
2630

2731
retries := 1

pkg/model/watchdog.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -540,7 +540,7 @@ func (wd *WatchDog) evictLRUModel() {
540540
wd.Unlock()
541541

542542
// Shutdown the model
543-
if err := wd.pm.ShutdownModel(lruModel.model); err != nil {
543+
if err := wd.pm.ShutdownModel(lruModel.model); err != nil && err != modelNotFoundErr {
544544
xlog.Error("[WatchDog] error shutting down model during memory reclamation", "error", err, "model", lruModel.model)
545545
} else {
546546
// Untrack the model

0 commit comments

Comments
 (0)