rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Don't do horde on local models

Browse Source
This commit is contained in:
Henk
2023-10-29 16:39:21 +01:00
parent 0896b12d53
commit 41604aeba6
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modeling/inference_models/horde/class.py
View File

@@ -39,13 +39,7 @@ class model_backend(InferenceModel):
self.capabilties = ModelCapabilities(api_host=False) self.capabilties = ModelCapabilities(api_host=False)
def is_valid(self, model_name, model_path, menu_path): def is_valid(self, model_name, model_path, menu_path):
try: return model_name == "CLUSTER"
if utils.koboldai_vars.horde_api_key is not "0000000000":
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]
except:
return False
def get_requested_parameters(self, model_name, model_path, menu_path, parameters = {}): def get_requested_parameters(self, model_name, model_path, menu_path, parameters = {}):
self.models = self.get_cluster_models() self.models = self.get_cluster_models()