liujch1998 commited on
Commit
9679701
·
1 Parent(s): 6477832

Add 3 olmo2 indexes

Browse files
Files changed (1) hide show
  1. constants.py +3 -0
constants.py CHANGED
@@ -2,6 +2,9 @@ import os
2
 
3
  # options
4
  INDEX_BY_DESC = {
 
 
 
5
  'Dolma-v1.7 (2.6T tokens)': 'v4_dolma-v1_7_llama',
6
  'RedPajama (1.4T tokens)': 'v4_rpj_llama_s4',
7
  'Pile-train (380B tokens)': 'v4_piletrain_llama',
 
2
 
3
  # options
4
  INDEX_BY_DESC = {
5
+ 'OLMo 2 32B Instruct (4.6T tokens)': 'v4_olmo-2-0325-32b-instruct_llama',
6
+ 'OLMo 2 13B Instruct (4.6T tokens)': 'v4_olmo-2-1124-13b-instruct_llama',
7
+ 'OLMoE 1B 7B Instruct (4.6T tokens)': 'v4_olmoe-0125-1b-7b-instruct_llama',
8
  'Dolma-v1.7 (2.6T tokens)': 'v4_dolma-v1_7_llama',
9
  'RedPajama (1.4T tokens)': 'v4_rpj_llama_s4',
10
  'Pile-train (380B tokens)': 'v4_piletrain_llama',