From 908dc8ea60f14c4267b1b3080ec094d1041f84a1 Mon Sep 17 00:00:00 2001 From: ebolam Date: Tue, 27 Sep 2022 15:59:56 -0400 Subject: [PATCH] Fix for older model loading --- aiserver.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/aiserver.py b/aiserver.py index 2003b532..bd41fb02 100644 --- a/aiserver.py +++ b/aiserver.py @@ -730,15 +730,7 @@ def getModelSelection(modellist): getModelSelection(mainmenu) def check_if_dir_is_model(path): - if os.path.exists(path): - try: - from transformers import AutoConfig - model_config = AutoConfig.from_pretrained(path) - except: - return False - return True - else: - return False + return os.path.exists(os.path.join(path, 'config.json')): #==================================================================# # Return all keys in tokenizer dictionary containing char