diff --git a/aiserver.py b/aiserver.py
index 1ff967b9..9995458c 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1514,8 +1514,10 @@ def get_model_info(model, directory=""):
         pass
     else:
         layer_count = get_layer_count(model, directory=directory)
+        print("Layer Count: {}".format(layer_count))
         if layer_count is None:
             breakmodel = False
+            gpu = True
         else:
             breakmodel = True
             if model in ["NeoCustom", "GPT2Custom"]: