Spaces:
Running
Running
Commit
Β·
5392172
1
Parent(s):
f25f82c
update
Browse files- src/utils.py +3 -3
src/utils.py
CHANGED
|
@@ -20,7 +20,7 @@ def change_tab(query_param):
|
|
| 20 |
LLM_MODEL_ARCHS = {
|
| 21 |
"stablelm_epoch": "π΄ StableLM-Epoch",
|
| 22 |
"stablelm_alpha": "π΄ StableLM-Alpha",
|
| 23 |
-
"mixformer-sequential": "
|
| 24 |
"RefinedWebModel": "π¦
Falcon",
|
| 25 |
"gpt_bigcode": "β StarCoder",
|
| 26 |
"RefinedWeb": "π¦
Falcon",
|
|
@@ -28,19 +28,19 @@ LLM_MODEL_ARCHS = {
|
|
| 28 |
"internlm": "π§βπ InternLM",
|
| 29 |
"mistral": "βοΈ Mistral",
|
| 30 |
"codegen": "βΎοΈ CodeGen",
|
|
|
|
| 31 |
"falcon": "π¦
Falcon",
|
| 32 |
"bloom": "πΈ Bloom",
|
| 33 |
"llama": "π¦ LLaMA",
|
|
|
|
| 34 |
"mpt": "π§± MPT",
|
| 35 |
"Yi": "δΊΊ Yi",
|
| 36 |
# suggest something
|
| 37 |
"gpt_neox": "GPT-NeoX",
|
| 38 |
"gpt_neo": "GPT-Neo",
|
| 39 |
-
"chatglm": "ChatGLM",
|
| 40 |
"gpt2": "GPT-2",
|
| 41 |
"gptj": "GPT-J",
|
| 42 |
"xglm": "XGLM",
|
| 43 |
-
"rwkv": "RWKV",
|
| 44 |
"bart": "BART",
|
| 45 |
"opt": "OPT",
|
| 46 |
}
|
|
|
|
| 20 |
LLM_MODEL_ARCHS = {
|
| 21 |
"stablelm_epoch": "π΄ StableLM-Epoch",
|
| 22 |
"stablelm_alpha": "π΄ StableLM-Alpha",
|
| 23 |
+
"mixformer-sequential": "π§βπ» Phi",
|
| 24 |
"RefinedWebModel": "π¦
Falcon",
|
| 25 |
"gpt_bigcode": "β StarCoder",
|
| 26 |
"RefinedWeb": "π¦
Falcon",
|
|
|
|
| 28 |
"internlm": "π§βπ InternLM",
|
| 29 |
"mistral": "βοΈ Mistral",
|
| 30 |
"codegen": "βΎοΈ CodeGen",
|
| 31 |
+
"chatglm": "π¬ ChatGLM",
|
| 32 |
"falcon": "π¦
Falcon",
|
| 33 |
"bloom": "πΈ Bloom",
|
| 34 |
"llama": "π¦ LLaMA",
|
| 35 |
+
"rwkv": "π¦ββ¬ RWKV",
|
| 36 |
"mpt": "π§± MPT",
|
| 37 |
"Yi": "δΊΊ Yi",
|
| 38 |
# suggest something
|
| 39 |
"gpt_neox": "GPT-NeoX",
|
| 40 |
"gpt_neo": "GPT-Neo",
|
|
|
|
| 41 |
"gpt2": "GPT-2",
|
| 42 |
"gptj": "GPT-J",
|
| 43 |
"xglm": "XGLM",
|
|
|
|
| 44 |
"bart": "BART",
|
| 45 |
"opt": "OPT",
|
| 46 |
}
|