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

Fixed Categories in Presets

Browse Source
This commit is contained in:
ebolam
2022-08-04 09:21:09 -04:00
parent 5da72fda84
commit 6060accbca
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
aiserver.py
View File

@@ -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)) 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 Size'] in koboldai_vars.model.replace("6.7B", "6B") and preset['uid'] not in used_ids:
if preset['Model Category'] == 'Custom': if preset['Model Category'] == 'Custom':
to_use['Recommended']['Custom'].append(preset) to_use['Same Class']['Custom'].append(preset)
else: else:
to_use['Recommended']['Official'].append(preset) to_use['Same Class']['Official'].append(preset)
used_ids.append(preset['uid']) used_ids.append(preset['uid'])
#Build the rest of the stuff #Build the rest of the stuff
for preset in presets: for preset in presets: