From bfc07073e3417593cd0a5525b9038317ce805143 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Mon, 7 Mar 2022 19:33:24 -0500
Subject: [PATCH] layer count fix

---
 aiserver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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