From fd7ba9f70e10ea7d6bebdab20f3140b270e25e96 Mon Sep 17 00:00:00 2001
From: henk717 <henk717@msn.com>
Date: Tue, 22 Feb 2022 19:22:08 +0100
Subject: [PATCH] Also check for Config in models/

---
 aiserver.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/aiserver.py b/aiserver.py
index d274cbb8..c5f8c1ec 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -548,6 +548,12 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly", "TPUMeshTransforme
             vars.model_type = model_config.model_type
         except ValueError as e:
             vars.model_type = "not_found"
+    elif(os.path.isdir("models/{}".format(vars.custmodpth.replace('/', '_')))):
+        try:
+            model_config = AutoConfig.from_pretrained("models/{}".format(vars.custmodpth.replace('/', '_')), cache_dir="cache/")
+            vars.model_type = model_config.model_type
+        except ValueError as e:
+            vars.model_type = "not_found"
     else:
         try:
             model_config = AutoConfig.from_pretrained(vars.custmodpth, cache_dir="cache/")