IlyasMoutawwakil HF staff commited on
Commit
5392172
β€’
1 Parent(s): f25f82c
Files changed (1) hide show
  1. 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": "Ο† Phi",
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
  }