IlyasMoutawwakil HF staff commited on
Commit
f25f82c
Β·
1 Parent(s): 9a3f7b4
Files changed (1) hide show
  1. 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",