layer count fix

2022-03-07 19:33:24 -05:00 · 2022-03-07 19:33:24 -05:00 · bfc07073e3
parent d8ab58892d
commit bfc07073e3
1 changed files with 1 additions and 1 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -821,7 +821,7 @@ def get_layer_count(model, directory=""):
        if model_config is None:
            return None
        try:
-            layers = model_config.num_layers
+            layers = model_config.num_layers if hasattr(model_config, "num_layers") else model_config.n_layer
        except:
            layers = None
            pass