rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Merge pull request #217 from ebolam/united

Browse Source 
Fix for older model loading
This commit is contained in:
henk717
2022-09-27 22:41:13 +02:00
committed by GitHub
parent 62921c4896 908dc8ea60
commit d772837ad0
1 changed files with 1 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
aiserver.py
View File

@@ -730,15 +730,7 @@ def getModelSelection(modellist):
getModelSelection(mainmenu) getModelSelection(mainmenu)
def check_if_dir_is_model(path): def check_if_dir_is_model(path):
if os.path.exists(path): return os.path.exists(os.path.join(path, 'config.json')):
try:
from transformers import AutoConfig
model_config = AutoConfig.from_pretrained(path)
except:
return False
return True
else:
return False
#==================================================================# #==================================================================#
# Return all keys in tokenizer dictionary containing char # Return all keys in tokenizer dictionary containing char