Commit
·
f25f82c
1
Parent(s):
9a3f7b4
update
Browse files- src/utils.py +1 -1
src/utils.py
CHANGED
|
@@ -25,6 +25,7 @@ LLM_MODEL_ARCHS = {
|
|
| 25 |
"gpt_bigcode": "⭐ StarCoder",
|
| 26 |
"RefinedWeb": "🦅 Falcon",
|
| 27 |
"baichuan": "🌊 Baichuan",
|
|
|
|
| 28 |
"mistral": "Ⓜ️ Mistral",
|
| 29 |
"codegen": "♾️ CodeGen",
|
| 30 |
"falcon": "🦅 Falcon",
|
|
@@ -34,7 +35,6 @@ LLM_MODEL_ARCHS = {
|
|
| 34 |
"Yi": "人 Yi",
|
| 35 |
# suggest something
|
| 36 |
"gpt_neox": "GPT-NeoX",
|
| 37 |
-
"internlm": "InternLM",
|
| 38 |
"gpt_neo": "GPT-Neo",
|
| 39 |
"chatglm": "ChatGLM",
|
| 40 |
"gpt2": "GPT-2",
|
|
|
|
| 25 |
"gpt_bigcode": "⭐ StarCoder",
|
| 26 |
"RefinedWeb": "🦅 Falcon",
|
| 27 |
"baichuan": "🌊 Baichuan",
|
| 28 |
+
"internlm": "🧑🎓 InternLM",
|
| 29 |
"mistral": "Ⓜ️ Mistral",
|
| 30 |
"codegen": "♾️ CodeGen",
|
| 31 |
"falcon": "🦅 Falcon",
|
|
|
|
| 35 |
"Yi": "人 Yi",
|
| 36 |
# suggest something
|
| 37 |
"gpt_neox": "GPT-NeoX",
|
|
|
|
| 38 |
"gpt_neo": "GPT-Neo",
|
| 39 |
"chatglm": "ChatGLM",
|
| 40 |
"gpt2": "GPT-2",
|