From 6060accbca3a81039be3e652199c9d80299807fe Mon Sep 17 00:00:00 2001 From: ebolam Date: Thu, 4 Aug 2022 09:21:09 -0400 Subject: [PATCH] Fixed Categories in Presets --- aiserver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aiserver.py b/aiserver.py index 3d974d4f..87f5e5a0 100644 --- a/aiserver.py +++ b/aiserver.py @@ -2192,9 +2192,9 @@ def load_model(use_gpu=True, gpu_layers=None, disk_layers=None, initial_load=Fal print("Found: {} - {} -> {}".format(preset['Model Size'] in koboldai_vars.model, preset['Model Size'], koboldai_vars.model)) if preset['Model Size'] in koboldai_vars.model.replace("6.7B", "6B") and preset['uid'] not in used_ids: if preset['Model Category'] == 'Custom': - to_use['Recommended']['Custom'].append(preset) + to_use['Same Class']['Custom'].append(preset) else: - to_use['Recommended']['Official'].append(preset) + to_use['Same Class']['Official'].append(preset) used_ids.append(preset['uid']) #Build the rest of the stuff for preset in presets: