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 |
}
|