diff --git a/aiserver.py b/aiserver.py
index 16b3b926..097acfa3 100644
--- 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