Always delete process

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
mudler · Nov 26, 2023 · de1abfc · de1abfc
1 parent 41fd3ef
commit de1abfc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/model/loader.go b/pkg/model/loader.go
@@ -154,11 +154,12 @@ func (ml *ModelLoader) ShutdownModel(modelName string) error {
 }
 
 func (ml *ModelLoader) StopModel(modelName string) error {
+	defer ml.deleteProcess(modelName)
 	if _, ok := ml.models[modelName]; !ok {
 		return fmt.Errorf("model %s not found", modelName)
 	}
-
-	return ml.deleteProcess(modelName)
+	return nil
+	//return ml.deleteProcess(modelName)
 }
 
 func (ml *ModelLoader) CheckIsLoaded(s string) ModelAddress {