Hugging Face's logo

allenai
/
OLMoE-1B-7B-0125-DPO

Text Generation
Transformers
PyTorch
Safetensors
English
olmoe
conversational
Model card Files
xet
Community
1
vwxyzjn commited on
Commit
abce207
·
verified ·
1 Parent(s): c4e7067

Upload folder using huggingface_hub

Browse files
Files changed (11) hide show
  1. README.md +215 -0
  2. config.json +33 -0
  3. generation_config.json +6 -0
  4. metadata.json +1 -0
  5. pytorch_model-00001-of-00003.bin +3 -0
  6. pytorch_model-00002-of-00003.bin +3 -0
  7. pytorch_model-00003-of-00003.bin +3 -0
  8. pytorch_model.bin.index.json +0 -0
  9. special_tokens_map.json +23 -0
  10. tokenizer.json +0 -0
  11. tokenizer_config.json +248 -0
README.md ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ model-index:
4
+ - name: allenai/open_instruct_dev
5
+ results:
6
+ - task:
7
+ type: preference_evaluation
8
+ dataset:
9
+ name: reward-bench
10
+ type: allenai/reward-bench
11
+ metrics:
12
+ - type: accuracy
13
+ value: 1.0
14
+ - type: accuracy
15
+ value: 1.0
16
+ - type: accuracy
17
+ value: 1.0
18
+ - type: accuracy
19
+ value: 1.0
20
+ ---
21
+
22
+ # Model Card for allenai/open_instruct_dev
23
+
24
+ <!-- Provide a quick summary of what the model is/does. -->
25
+
26
+
27
+
28
+ ## Model Details
29
+
30
+ ### Model Description
31
+
32
+ <!-- Provide a longer summary of what this model is. -->
33
+
34
+
35
+
36
+ - **Developed by:** [More Information Needed]
37
+ - **Funded by [optional]:** [More Information Needed]
38
+ - **Shared by [optional]:** [More Information Needed]
39
+ - **Model type:** [More Information Needed]
40
+ - **Language(s) (NLP):** en
41
+ - **License:** [More Information Needed]
42
+ - **Finetuned from model [optional]:** [More Information Needed]
43
+
44
+ ### Model Sources [optional]
45
+
46
+ <!-- Provide the basic links for the model. -->
47
+
48
+ - **Repository:** [More Information Needed]
49
+ - **Paper [optional]:** [More Information Needed]
50
+ - **Demo [optional]:** [More Information Needed]
51
+
52
+ ## Uses
53
+
54
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
55
+
56
+ ### Direct Use
57
+
58
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
59
+
60
+ [More Information Needed]
61
+
62
+ ### Downstream Use [optional]
63
+
64
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
65
+
66
+ [More Information Needed]
67
+
68
+ ### Out-of-Scope Use
69
+
70
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
71
+
72
+ [More Information Needed]
73
+
74
+ ## Bias, Risks, and Limitations
75
+
76
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
77
+
78
+ [More Information Needed]
79
+
80
+ ### Recommendations
81
+
82
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
83
+
84
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
85
+
86
+ ## How to Get Started with the Model
87
+
88
+ Use the code below to get started with the model.
89
+
90
+ [More Information Needed]
91
+
92
+ ## Training Details
93
+
94
+ ### Training Data
95
+
96
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
97
+
98
+ [More Information Needed]
99
+
100
+ ### Training Procedure
101
+
102
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
103
+
104
+ #### Preprocessing [optional]
105
+
106
+ [More Information Needed]
107
+
108
+
109
+ #### Training Hyperparameters
110
+
111
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
112
+
113
+ #### Speeds, Sizes, Times [optional]
114
+
115
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
116
+
117
+ [More Information Needed]
118
+
119
+ ## Evaluation
120
+
121
+ <!-- This section describes the evaluation protocols and provides the results. -->
122
+
123
+ ### Testing Data, Factors & Metrics
124
+
125
+ #### Testing Data
126
+
127
+ <!-- This should link to a Dataset Card if possible. -->
128
+
129
+ [More Information Needed]
130
+
131
+ #### Factors
132
+
133
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
134
+
135
+ [More Information Needed]
136
+
137
+ #### Metrics
138
+
139
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
140
+
141
+ [More Information Needed]
142
+
143
+ ### Results
144
+
145
+ [More Information Needed]
146
+
147
+ #### Summary
148
+
149
+
150
+
151
+ ## Model Examination [optional]
152
+
153
+ <!-- Relevant interpretability work for the model goes here -->
154
+
155
+ [More Information Needed]
156
+
157
+ ## Environmental Impact
158
+
159
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
160
+
161
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
162
+
163
+ - **Hardware Type:** [More Information Needed]
164
+ - **Hours used:** [More Information Needed]
165
+ - **Cloud Provider:** [More Information Needed]
166
+ - **Compute Region:** [More Information Needed]
167
+ - **Carbon Emitted:** [More Information Needed]
168
+
169
+ ## Technical Specifications [optional]
170
+
171
+ ### Model Architecture and Objective
172
+
173
+ [More Information Needed]
174
+
175
+ ### Compute Infrastructure
176
+
177
+ [More Information Needed]
178
+
179
+ #### Hardware
180
+
181
+ [More Information Needed]
182
+
183
+ #### Software
184
+
185
+ [More Information Needed]
186
+
187
+ ## Citation [optional]
188
+
189
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
190
+
191
+ **BibTeX:**
192
+
193
+ [More Information Needed]
194
+
195
+ **APA:**
196
+
197
+ [More Information Needed]
198
+
199
+ ## Glossary [optional]
200
+
201
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
202
+
203
+ [More Information Needed]
204
+
205
+ ## More Information [optional]
206
+
207
+ [More Information Needed]
208
+
209
+ ## Model Card Authors [optional]
210
+
211
+ [More Information Needed]
212
+
213
+ ## Model Card Contact
214
+
215
+ [More Information Needed]
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "allenai/open_instruct_dev",
3
+ "architectures": [
4
+ "OlmoeForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "clip_qkv": null,
9
+ "eos_token_id": 50279,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 2048,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 1024,
14
+ "max_position_embeddings": 4096,
15
+ "model_type": "olmoe",
16
+ "norm_topk_prob": false,
17
+ "num_attention_heads": 16,
18
+ "num_experts": 64,
19
+ "num_experts_per_tok": 8,
20
+ "num_hidden_layers": 16,
21
+ "num_key_value_heads": 16,
22
+ "output_router_logits": false,
23
+ "pad_token_id": 1,
24
+ "rms_norm_eps": 1e-05,
25
+ "rope_scaling": null,
26
+ "rope_theta": 10000.0,
27
+ "router_aux_loss_coef": 0.01,
28
+ "tie_word_embeddings": false,
29
+ "torch_dtype": "bfloat16",
30
+ "transformers_version": "4.47.0.dev0",
31
+ "use_cache": true,
32
+ "vocab_size": 50304
33
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "eos_token_id": 50279,
4
+ "pad_token_id": 1,
5
+ "transformers_version": "4.47.0.dev0"
6
+ }
metadata.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_name": "0119_node4_dpo_olmoe_1blr_9e-7", "model_type": "dpo", "datasets": ["allenai/tulu-3-pref-personas-instruction-following", "ai2-adapt-dev/wildchat_v3.9_used_on_policy_prompts-olmoe", "ai2-adapt-dev/wildchat_v3.9_unused_off_policy_prompts-olmoe", "ai2-adapt-dev/DaringAnteater-preferences-olmoe", "ai2-adapt-dev/uf_cleaned-olmoe", "ai2-adapt-dev/IF_Taxonomy-olmoe", "ai2-adapt-dev/sft_v3.9_used_off_policy_prompts-olmoe", "ai2-adapt-dev/sft_v3.9_used_on_policy_prompts-olmoe", "ai2-adapt-dev/Wildchat-prefs-280824-olmoe"], "base_model": "allenai/open_instruct_dev", "wandb_path": "https://wandb.ai/ai2-llm/open_instruct_internal/runs/mtbhb8vm", "beaker_experiment": "https://beaker.org/ex/01JJC8QSMJBCZYAX35JBBC0JPP/", "beaker_datasets": ["https://beaker.org/ds/01JJC8QSP21RYMG0G6RJM685JG", "https://beaker.org/ds/01JJC8QSV12E4TCGSZ8AG5EQQT", "https://beaker.org/ds/01JJC8QSZRRJHNR8DXMP814DBD", "https://beaker.org/ds/01JJC8QT51B9KTE8XN0D7ZX502", "https://beaker.org/ds/01JJCAGG9GEEJ4HGNX184EB9Q9", "https://beaker.org/ds/01JJCAGGECE73JG9BZPQPAJ7WH", "https://beaker.org/ds/01JJCAGGJRDM2S6NXDTEF3T338", "https://beaker.org/ds/01JJCAGGQA0QKSBNAWQWTZQ6X6"]}
pytorch_model-00001-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f7f74ef3ccf88d8a79dad991ea07f857c8a0039bd0821c1ef94141fe4eee3bc
3
+ size 4998004590
pytorch_model-00002-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:950d6e0b061a1facda62e62855e184ecfad0daaf70dbf813fb27aefd56a22e79
3
+ size 4997506338
pytorch_model-00003-of-00003.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15b93b7eb2e527683352e4b4c5a52c7df6b5414e87687d69c20d870879ef9c3c
3
+ size 3843938359
pytorch_model.bin.index.json ADDED
The diff for this file is too large to render. See raw diff
 
special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "|||IP_ADDRESS|||",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "|||IP_ADDRESS|||",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<pad>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,248 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<|padding|>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "50254": {
23
+ "content": " ",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "50255": {
31
+ "content": " ",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "50256": {
39
+ "content": " ",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "50257": {
47
+ "content": " ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "50258": {
55
+ "content": " ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "50259": {
63
+ "content": " ",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "50260": {
71
+ "content": " ",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "50261": {
79
+ "content": " ",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "50262": {
87
+ "content": " ",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "50263": {
95
+ "content": " ",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "50264": {
103
+ "content": " ",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "50265": {
111
+ "content": " ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "50266": {
119
+ "content": " ",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ },
126
+ "50267": {
127
+ "content": " ",
128
+ "lstrip": false,
129
+ "normalized": true,
130
+ "rstrip": false,
131
+ "single_word": false,
132
+ "special": false
133
+ },
134
+ "50268": {
135
+ "content": " ",
136
+ "lstrip": false,
137
+ "normalized": true,
138
+ "rstrip": false,
139
+ "single_word": false,
140
+ "special": false
141
+ },
142
+ "50269": {
143
+ "content": " ",
144
+ "lstrip": false,
145
+ "normalized": true,
146
+ "rstrip": false,
147
+ "single_word": false,
148
+ "special": false
149
+ },
150
+ "50270": {
151
+ "content": " ",
152
+ "lstrip": false,
153
+ "normalized": true,
154
+ "rstrip": false,
155
+ "single_word": false,
156
+ "special": false
157
+ },
158
+ "50271": {
159
+ "content": " ",
160
+ "lstrip": false,
161
+ "normalized": true,
162
+ "rstrip": false,
163
+ "single_word": false,
164
+ "special": false
165
+ },
166
+ "50272": {
167
+ "content": " ",
168
+ "lstrip": false,
169
+ "normalized": true,
170
+ "rstrip": false,
171
+ "single_word": false,
172
+ "special": false
173
+ },
174
+ "50273": {
175
+ "content": " ",
176
+ "lstrip": false,
177
+ "normalized": true,
178
+ "rstrip": false,
179
+ "single_word": false,
180
+ "special": false
181
+ },
182
+ "50274": {
183
+ "content": " ",
184
+ "lstrip": false,
185
+ "normalized": true,
186
+ "rstrip": false,
187
+ "single_word": false,
188
+ "special": false
189
+ },
190
+ "50275": {
191
+ "content": " ",
192
+ "lstrip": false,
193
+ "normalized": true,
194
+ "rstrip": false,
195
+ "single_word": false,
196
+ "special": false
197
+ },
198
+ "50276": {
199
+ "content": " ",
200
+ "lstrip": false,
201
+ "normalized": true,
202
+ "rstrip": false,
203
+ "single_word": false,
204
+ "special": false
205
+ },
206
+ "50277": {
207
+ "content": "|||EMAIL_ADDRESS|||",
208
+ "lstrip": false,
209
+ "normalized": true,
210
+ "rstrip": false,
211
+ "single_word": false,
212
+ "special": false
213
+ },
214
+ "50278": {
215
+ "content": "|||PHONE_NUMBER|||",
216
+ "lstrip": false,
217
+ "normalized": true,
218
+ "rstrip": false,
219
+ "single_word": false,
220
+ "special": false
221
+ },
222
+ "50279": {
223
+ "content": "|||IP_ADDRESS|||",
224
+ "lstrip": false,
225
+ "normalized": true,
226
+ "rstrip": false,
227
+ "single_word": false,
228
+ "special": true
229
+ },
230
+ "50280": {
231
+ "content": "<pad>",
232
+ "lstrip": false,
233
+ "normalized": false,
234
+ "rstrip": false,
235
+ "single_word": false,
236
+ "special": true
237
+ }
238
+ },
239
+ "bos_token": "|||IP_ADDRESS|||",
240
+ "chat_template": "{{ bos_token }}{% for message in messages %}{% if message['role'] == 'system' %}{{ '<|system|>\n' + message['content'] + '\n' }}{% elif message['role'] == 'user' %}{{ '<|user|>\n' + message['content'] + '\n' }}{% elif message['role'] == 'assistant' %}{% if not loop.last %}{{ '<|assistant|>\n' + message['content'] + eos_token + '\n' }}{% else %}{{ '<|assistant|>\n' + message['content'] + eos_token }}{% endif %}{% endif %}{% if loop.last and add_generation_prompt %}{{ '<|assistant|>\n' }}{% endif %}{% endfor %}",
241
+ "clean_up_tokenization_spaces": false,
242
+ "eos_token": "|||IP_ADDRESS|||",
243
+ "extra_special_tokens": {},
244
+ "model_max_length": 1000000000000000019884624838656,
245
+ "pad_token": "<pad>",
246
+ "tokenizer_class": "GPTNeoXTokenizer",
247
+ "unk_token": null
248
+ }