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 error due to breakmodel file being partially empty

Browse Source
This commit is contained in:
ebolam
2022-09-01 08:59:42 -04:00
parent 57719bdcd2
commit 1f6eefa6df
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@@ -1352,6 +1352,8 @@ def get_model_info(model, directory=""):
break_values = break_values.split(",") break_values = break_values.split(",")
else: else:
break_values = [layer_count] break_values = [layer_count]
if break_values == ['']:
break_values = []
break_values = [int(x) for x in break_values] break_values = [int(x) for x in break_values]
break_values += [0] * (gpu_count - len(break_values)) break_values += [0] * (gpu_count - len(break_values))
emit('from_server', {'cmd': 'selected_model_info', 'key_value': key_value, 'key':key, emit('from_server', {'cmd': 'selected_model_info', 'key_value': key_value, 'key':key,