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

Fix for model loading from paths

Browse Source
This commit is contained in:
ebolam
2023-06-13 20:05:50 -04:00
parent e2801fb5c1
commit abe07a2e95
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
aiserver.py
View File

@@ -6234,8 +6234,7 @@ def UI_2_select_model(data):
else:
#Here we have a model that's not in our menu structure (either a custom model or a custom path
#so we'll just go through all the possible loaders
for model_backend_type in model_backends:
for model_backend in model_backend_type_crosswalk[model_backend_type]:
for model_backend in model_backends:
if model_backends[model_backend].is_valid(data["name"], data["path"] if 'path' in data else None, data["menu"]):
valid_loaders[model_backend] = model_backends[model_backend].get_requested_parameters(data["name"], data["path"] if 'path' in data else None, data["menu"])
emit("selected_model_info", {"model_backends": valid_loaders})
@@ -6245,8 +6244,7 @@ def UI_2_select_model(data):
output = []
for path in paths:
valid=False
for model_backend_type_crosswalk in model_backends:
for model_backend in model_backend_type_crosswalk[model_backend_type]:
for model_backend in model_backends:
if model_backends[model_backend].is_valid(path[1], path[0], "Custom"):
logger.debug("{} says valid".format(model_backend))
valid=True