From 9501dd34eaa65d6a4b525bd38a4aa204a564927c Mon Sep 17 00:00:00 2001 From: ebolam Date: Fri, 6 Oct 2023 09:24:59 -0400 Subject: [PATCH] Fix for aiserver asking horde for model list on start of koboldai. This meant that every run of KoboldAI resulted in horde being effectively notified --- modeling/inference_models/horde/class.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modeling/inference_models/horde/class.py b/modeling/inference_models/horde/class.py index 09ba71d5..8c05c688 100644 --- a/modeling/inference_models/horde/class.py +++ b/modeling/inference_models/horde/class.py @@ -29,7 +29,7 @@ class model_backend(InferenceModel): super().__init__() self.url = "https://horde.koboldai.net" self.key = "0000000000" - self.models = self.get_cluster_models() + self.models = [] self.model_name = "Horde" self.model = [] self.request_id = None @@ -39,10 +39,12 @@ class model_backend(InferenceModel): self.capabilties = ModelCapabilities(api_host=False) def is_valid(self, model_name, model_path, menu_path): + self.models = self.get_cluster_models() logger.debug("Horde Models: {}".format(self.models)) return model_name == "CLUSTER" or model_name in [x['value'] for x in self.models] def get_requested_parameters(self, model_name, model_path, menu_path, parameters = {}): + self.models = self.get_cluster_models() if os.path.exists("settings/horde.model_backend.settings") and 'base_url' not in vars(self): with open("settings/horde.model_backend.settings", "r") as f: temp = json.load(f)