diff --git a/llmengine/base_chat_llm.py b/llmengine/base_chat_llm.py index 12b7438..54954ac 100644 --- a/llmengine/base_chat_llm.py +++ b/llmengine/base_chat_llm.py @@ -15,7 +15,8 @@ def llm_register(model_key, Klass): model_pathMap[model_key] = Klass def get_llm_class(model_path): - keys = [k for k in model_pathMap.keys()].sort() + keys = [k for k in model_pathMap.keys()] + keys.sort() keys.reverse() for k in keys: if len(model_path.split(k)) > 1: