rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-02-17 20:20:45 +01:00
Code Issues Projects Releases Wiki Activity

layer count fix

Browse Source
This commit is contained in:
ebolam 2022-03-07 19:33:24 -05:00
parent d8ab58892d
commit bfc07073e3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -821,7 +821,7 @@ def get_layer_count(model, directory=""):
if model_config is None: if model_config is None:
return None return None
try: try:
layers = model_config.num_layers layers = model_config.num_layers if hasattr(model_config, "num_layers") else model_config.n_layer
except: except:
layers = None layers = None
pass pass