internlm-20b的仓库名称为internlm-chat-20b (#713)

This commit is contained in:
FangYin Cheng
2023-10-26 15:29:02 +08:00
committed by GitHub

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 # 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": os.path.join(MODEL_PATH, "internlm-chat-7b"),
"internlm-7b-8k": os.path.join(MODEL_PATH, "internlm-chat-7b-8k"), "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 # For test now
"opt-125m": os.path.join(MODEL_PATH, "opt-125m"), "opt-125m": os.path.join(MODEL_PATH, "opt-125m"),
} }