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

22
aiserver.py
View File

@@ -6234,10 +6234,9 @@ def UI_2_select_model(data):
else: else:
#Here we have a model that's not in our menu structure (either a custom model or a custom path #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 #so we'll just go through all the possible loaders
for model_backend_type in model_backends: for model_backend in model_backends:
for model_backend in model_backend_type_crosswalk[model_backend_type]: if model_backends[model_backend].is_valid(data["name"], data["path"] if 'path' in data else None, data["menu"]):
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"])
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}) emit("selected_model_info", {"model_backends": valid_loaders})
else: else:
#Get directories #Get directories
@@ -6245,14 +6244,13 @@ def UI_2_select_model(data):
output = [] output = []
for path in paths: for path in paths:
valid=False valid=False
for model_backend_type_crosswalk in model_backends: for model_backend in model_backends:
for model_backend in model_backend_type_crosswalk[model_backend_type]: if model_backends[model_backend].is_valid(path[1], path[0], "Custom"):
if model_backends[model_backend].is_valid(path[1], path[0], "Custom"): logger.debug("{} says valid".format(model_backend))
logger.debug("{} says valid".format(model_backend)) valid=True
valid=True break
break else:
else: logger.debug("{} says invalid".format(model_backend))
logger.debug("{} says invalid".format(model_backend))
output.append({'label': path[1], 'name': path[1], 'size': "", "menu": "Custom", 'path': path[0], 'isMenu': not valid}) output.append({'label': path[1], 'name': path[1], 'size': "", "menu": "Custom", 'path': path[0], 'isMenu': not valid})
emit("open_model_load_menu", {"items": output+[{'label': 'Return to Main Menu', 'name':'mainmenu', 'size': "", "menu": "Custom", 'isMenu': True}], 'breadcrumbs': breadcrumbs}) emit("open_model_load_menu", {"items": output+[{'label': 'Return to Main Menu', 'name':'mainmenu', 'size': "", "menu": "Custom", 'isMenu': True}], 'breadcrumbs': breadcrumbs})