internlm-20b的仓库名称为internlm-chat-20b

This commit is contained in:
chengxi 2023-10-21 14:52:19 +08:00 committed by GitHub
parent 6c1a912555
commit 93a1ae5f6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -77,7 +77,7 @@ LLM_MODEL_CONFIG = {
# https://huggingface.co/internlm/internlm-chat-7b-v1_1, 7b vs 7b-v1.1: https://github.com/InternLM/InternLM/issues/288
"internlm-7b": os.path.join(MODEL_PATH, "internlm-chat-7b"),
"internlm-7b-8k": os.path.join(MODEL_PATH, "internlm-chat-7b-8k"),
"internlm-20b": os.path.join(MODEL_PATH, "internlm-20b-chat"),
"internlm-20b": os.path.join(MODEL_PATH, "internlm-chat-20b"),
# For test now
"opt-125m": os.path.join(MODEL_PATH, "opt-125m"),
}