Duplicate from allenai/OLMo-7B-Instruct

Co-authored-by: Hamish Ivison <hamishivi@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,184 @@
+---
+license: apache-2.0
+datasets:
+- allenai/dolma
+- allenai/tulu-v2-sft-mixture
+- allenai/ultrafeedback_binarized_cleaned
+language:
+- en
+---
+
+
+<img src="https://allenai.org/olmo/olmo-7b-animation.gif" alt="OLMo Logo" width="800" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
+
+
+# Model Card for OLMo 7B Instruct
+
+**For transformers versions v4.40.0 or newer, we suggest using [OLMo 7B Instruct HF](https://huggingface.co/allenai/OLMo-7B-Instruct-hf) instead.**
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+OLMo is a series of **O**pen **L**anguage **Mo**dels designed to enable the science of language models.
+The OLMo base models are trained on the [Dolma](https://huggingface.co/datasets/allenai/dolma) dataset.
+The adapted versions are trained on the [Tulu SFT mixture](https://huggingface.co/datasets/allenai/tulu-v2-sft-mixture) and, for the Instruct version, a [cleaned version of the UltraFeedback dataset](https://huggingface.co/datasets/allenai/ultrafeedback_binarized_cleaned).
+We release all code, checkpoints, logs (coming soon), and details involved in training these models.
+
+OLMo 7B Instruct and OLMo SFT are two adapted versions of these models trained for better question answering.
+They show the performance gain that OLMo base models can achieve with existing fine-tuning techniques.
+
+*Note:* This model requires installing `ai2-olmo` with pip and using `ai2-olmo`>=0.3.0 or HuggingFace Transformers<=4.39. New versions of the model will be released soon with compatibility improvements.
+
+**Note that this is the instruct variant of the original OLMo release - there are newer (better) versions of OLMo out now! Check [here](https://huggingface.co/collections/allenai/olmo-suite-65aeaae8fe5b6b2122b46778) for newer releases.**
+
+## Model Details
+
+We release two adapted model versions:
+The base models related to this adapted model are the following: 
+| Model | Training Method(s) | Datasets | Context Length | 
+|------|--------|---------|--|
+| [OLMo 7B SFT](https://huggingface.co/allenai/OLMo-7B-SFT)   | SFT | [Tulu 2 SFT Mix](https://huggingface.co/datasets/allenai/tulu-v2-sft-mixture)     | 2048        | 
+| [OLMo 7B Instruct](https://huggingface.co/allenai/OLMo-7B-Instruct) | SFT + DPO   | [Tulu 2 SFT Mix](https://huggingface.co/datasets/allenai/tulu-v2-sft-mixture) + [Ultrafeedback Cleaned](https://huggingface.co/datasets/allenai/ultrafeedback_binarized_cleaned)     | 2048 |
+
+
+The base models related to this adapted model are the following: 
+| Size | Training Tokens | Layers | Hidden Size | Attention Heads | Context Length |
+|------|--------|---------|-------------|-----------------|----------------|
+| [OLMo 1B](https://huggingface.co/allenai/OLMo-1B)   | 3 Trillion |16     | 2048        | 16              | 2048  |
+| [OLMo 7B](https://huggingface.co/allenai/OLMo-7B) | 2.5 Trillion   | 32     | 4096        | 32              |  2048  |
+| [OLMo 7B Twin 2T](https://huggingface.co/allenai/OLMo-7B-Twin-2T) | 2 Trillion   | 32     | 4096        | 32              |  2048  |
+
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+- **Developed by:** Allen Institute for AI (AI2)
+- **Supported by:** Databricks, Kempner Institute for the Study of Natural and Artificial Intelligence at Harvard University, AMD, CSC (Lumi Supercomputer), UW
+- **Model type:** a Transformer style autoregressive language model.
+- **Language(s) (NLP):** English
+- **License:** The code and model are released under Apache 2.0.
+- **Contact:** Technical inquiries: `olmo at allenai dot org`. Press: `press at allenai dot org`
+- **Date cutoff:** Feb./March 2023 based on Dolma dataset version.
+
+
+### Model Sources
+
+<!-- Provide the basic links for the model. -->
+
+- **Project Page:** https://allenai.org/olmo
+- **Repositories:** 
+    - Core repo (training, inference, fine-tuning etc.): https://github.com/allenai/OLMo
+    - Evaluation code: https://github.com/allenai/OLMo-Eval
+    - Further fine-tuning code: https://github.com/allenai/open-instruct
+- **Paper:** [Link](https://arxiv.org/abs/2402.00838)
+- **Technical blog post:** https://blog.allenai.org/olmo-open-language-model-87ccfc95f580
+- **W&B Logs:** https://wandb.ai/ai2-llm/OLMo-7B/reports/OLMo-7B--Vmlldzo2NzQyMzk5
+<!-- - **Press release:** TODO -->
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Inference
+Quickly get inference running with the following required installation:
+```bash
+pip install ai2-olmo
+```
+Now, proceed as usual with HuggingFace:
+```python
+from hf_olmo import OLMoForCausalLM, OLMoTokenizerFast
+olmo = OLMoForCausalLM.from_pretrained("allenai/OLMo-7B-Instruct")
+tokenizer = OLMoTokenizerFast.from_pretrained("allenai/OLMo-7B-Instruct")
+chat = [
+    { "role": "user", "content": "What is language modeling?" },
+]
+prompt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
+inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
+# optional verifying cuda
+# inputs = {k: v.to('cuda') for k,v in inputs.items()}
+# olmo = olmo.to('cuda')
+response = olmo.generate(input_ids=inputs.to(olmo.device), max_new_tokens=100, do_sample=True, top_k=50, top_p=0.95)
+print(tokenizer.batch_decode(response, skip_special_tokens=True)[0])
+>> '<|user|>\nWhat is language modeling?\n<|assistant|>\nLanguage modeling is a type of natural language processing (NLP) task or machine learning task that...'
+```
+
+You can make this slightly faster by quantizing the model, e.g. `OLMoForCausalLM.from_pretrained("allenai/OLMo-7B-Instruct", torch_dtype=torch.float16, load_in_8bit=True)` (requires `bitsandbytes`).
+The quantized model is more sensitive to typing / cuda, so it is recommended to pass the inputs as `inputs.input_ids.to('cuda')` to avoid potential issues.
+
+Note, you may see the following error if `ai2-olmo` is not installed correctly, which is caused by internal Python check naming. We'll update the code soon to make this error clearer.
+```bash
+    raise ImportError(
+ImportError: This modeling file requires the following packages that were not found in your environment: hf_olmo. Run `pip install hf_olmo`
+```
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+Core model results for the 7B adapted models are found below.
+
+| Model                 | MMLU 0-shot ↑ | AlpacaEval %win ↑ | ToxiGen % Toxic ↓ | TruthfulQA %Info+True ↑ |
+|-----------------------|---------------|--------------------|--------------------|-------------------------|
+| **OLMo (base)**      | 28.3          | -                  | 81.4               | 31.6                    |
+| MPT Chat          | 33.8          | 46.8               | 0.1                | 42.7                    |
+| Falcon Instruct   | 25.2          | 14.0               | 70.7               | 27.2                    |
+| RPJ-INCITE Chat   | 27.0          | 38.0               | 46.4               | 53.0                    |
+| Llama-2-Chat 7B      | 46.8          | 87.3               | 0.0                | 26.3                    |
+| AI2 Tulu 2 7B         | 50.4          | 73.9               | 7.0                | 51.7                    |
+| AI2 Tulu 2 7B DPO        | 50.7          | 85.1               | 0.5                | - *                     |
+| **[OLMo 7B SFT](https://huggingface.co/allenai/OLMo-7B-SFT)**         | 47.3          | 57.0               | 14.4               | 41.2                    |
+| **[OLMo 7B Instruct](https://huggingface.co/allenai/OLMo-7B-Instruct)**     | 46.2          | 69.3               | 1.7                | 52.0                    |
+
+*Following Ivison et al. 2023, we do not report Tulu 2 TruthfulQA scores due to test set contamination.
+## Model Details
+
+### Data
+For training data details, please see the [Dolma](https://huggingface.co/datasets/allenai/dolma), [Tulu 2](https://huggingface.co/datasets/allenai/tulu-v2-sft-mixture), and [UltraFeedback](https://huggingface.co/datasets/openbmb/UltraFeedback) documentation.
+
+### Architecture
+
+
+### Hyperparameters 
+
+The hyperparameters for the two phases of training are below:
+
+|                         | Learning Rate | Beta | Epochs | Warmup                                                                 | Weight Decay | Gradient Clipping | Maximum Sequence Length |
+|-------------------------|---------------|------|--------|------------------------------------------------------------------------|--------------|-------------------|-------------------------|
+| **SFT**                 | 2 × 10^-6     | N/A  | 3      | Linear warmup for the first 3% of total training time, then cooldown to 0 | 0            | 0                 | 2048                    |
+| **DPO**                 | 5 × 10^-7     | 0.1  | 3      | Linear warmup for the first 10% of total training time, then cooldown to 0| 0            | 0                 | 2048                    |
+
+Compared to Tulu 2, DPO hyperparameters are the same. SFT is lower LR and 3 epochs instead of 2 (and 2k length instead of 8k).
+
+## Bias, Risks, and Limitations
+
+This adapted OLMo model is a research artifact. 
+It is intended to benefit the research community interested in understanding the safety properties of LLMs and developers building safety tools for LLMs. 
+For this reason, the model does not include a specific safety filter or safety training data. 
+While our model scores well relative to its peers on ToxiGen, it is possible for the model to generate harmful and sensitive content from some user prompts. 
+We recommend developers exercise caution and consider the risks of the applications of this technology. 
+Furthermore, developers should consider implementing safeguards for biases, privacy, and other potential harms when appropriate. 
+Finally, as with every LLM, OLMo may produce factual-sounding outputs that may not be true, so developers and users are encouraged to confirm such outputs before relying on them. 
+All users of this model are responsible for how they use the model.
+
+
+## Citation
+
+**BibTeX:**
+
+```
+@article{Groeneveld2023OLMo,
+  title={OLMo: Accelerating the Science of Language Models},
+  author={Groeneveld, Dirk and Beltagy, Iz and Walsh, Pete and Bhagia, Akshita and Kinney, Rodney and Tafjord, Oyvind and Jha, Ananya Harsh and Ivison, Hamish and Magnusson, Ian and Wang, Yizhong and Arora, Shane and Atkinson, David and Authur, Russell and Chandu, Khyathi and Cohan, Arman and Dumas, Jennifer and Elazar, Yanai and Gu, Yuling and Hessel, Jack and Khot, Tushar and Merrill, William and Morrison, Jacob and Muennighoff, Niklas and Naik, Aakanksha and Nam, Crystal and Peters, Matthew E. and Pyatkin, Valentina and Ravichander, Abhilasha and Schwenk, Dustin and Shah, Saurabh and Smith, Will and Subramani, Nishant and Wortsman, Mitchell and Dasigi, Pradeep and Lambert, Nathan and Richardson, Kyle and Dodge, Jesse and Lo, Kyle and Soldaini, Luca and Smith, Noah A. and Hajishirzi, Hannaneh},
+  journal={Preprint},
+  year={2024}
+}
+```
+
+**APA:**
+
+Groeneveld, D., Beltagy, I., Walsh, P., Bhagia, A., Kinney, R., Tafjord, O., Jha, A., Ivison, H., Magnusson, I., Wang, Y., Arora, S., Atkinson, D., Authur, R., Chandu, K., Cohan, A., Dumas, J., Elazar, Y., Gu, Y., Hessel, J., Khot, T., Merrill, W., Morrison, J., Muennighoff, N., Naik, A., Nam, C., Peters, M., Pyatkin, V., Ravichander, A., Schwenk, D., Shah, S., Smith, W., Subramani, N., Wortsman, M., Dasigi, P., Lambert, N., Richardson, K., Dodge, J., Lo, K., Soldaini, L., Smith, N., & Hajishirzi, H. (2024). OLMo: Accelerating the Science of Language Models. Preprint.
+
+## Model Card Contact
+
+
+For errors in this model card, contact Nathan or Jacob, `{nathanl, jacobm} at allenai dot org`.

config.json

@@ -0,0 +1,52 @@
+{
+  "activation_type": "swiglu",
+  "alibi": false,
+  "alibi_bias_max": 8.0,
+  "architectures": [
+    "OLMoForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "attention_layer_norm": false,
+  "attention_layer_norm_with_affine": false,
+  "auto_map": {
+    "AutoConfig": "configuration_olmo.OLMoConfig",
+    "AutoModelForCausalLM": "modeling_olmo.OLMoForCausalLM",
+    "AutoTokenizer": [
+      "tokenization_olmo_fast.OLMoTokenizerFast",
+      "tokenization_olmo_fast.OLMoTokenizerFast"
+    ]
+  },
+  "bias_for_layer_norm": false,
+  "block_group_size": 1,
+  "block_type": "sequential",
+  "d_model": 4096,
+  "embedding_dropout": 0.0,
+  "embedding_size": 50304,
+  "eos_token_id": 50279,
+  "flash_attention": true,
+  "include_bias": false,
+  "init_cutoff_factor": null,
+  "init_device": "meta",
+  "init_fn": "mitchell",
+  "init_std": 0.02,
+  "layer_norm_type": "default",
+  "layer_norm_with_affine": false,
+  "max_sequence_length": 2048,
+  "mlp_hidden_size": 22016,
+  "mlp_ratio": 4,
+  "model_type": "hf_olmo",
+  "multi_query_attention": false,
+  "n_heads": 32,
+  "n_layers": 32,
+  "pad_token_id": 1,
+  "precision": "amp_bf16",
+  "residual_dropout": 0.0,
+  "rope": true,
+  "rope_full_precision": true,
+  "scale_logits": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.35.0.dev0",
+  "use_cache": true,
+  "vocab_size": 50280,
+  "weight_tying": false
+}

configuration_olmo.py

@@ -0,0 +1,3 @@
+from hf_olmo import OLMoConfig
+
+

model-00001-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:148acfce238941760f81b0d66228af4d711331102d7dc54af2c32deb0d07b6d4
+size 9945755032

model-00002-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec7997b27142e6a09aa705e280bff475322f5e6102dd25dc895ef5bda92b50b9
+size 3830452280

model.safetensors.index.json

@@ -0,0 +1,137 @@
+{
+    "metadata": {
+        "total_size": 13776191488
+    },
+    "weight_map": {
+        "model.transformer.blocks.0.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.0.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.0.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.0.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.1.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.1.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.1.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.1.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.10.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.10.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.10.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.10.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.11.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.11.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.11.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.11.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.12.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.12.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.12.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.12.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.13.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.13.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.13.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.13.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.14.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.14.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.14.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.14.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.15.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.15.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.15.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.15.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.16.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.16.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.16.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.16.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.17.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.17.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.17.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.17.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.18.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.18.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.18.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.18.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.19.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.19.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.19.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.19.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.2.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.2.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.2.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.2.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.20.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.20.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.20.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.20.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.21.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.21.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.21.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.21.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.22.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.22.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.22.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.22.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.23.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.23.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.23.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.23.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.24.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.24.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.24.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.24.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.25.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.25.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.25.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.25.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.26.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.26.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.26.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.26.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.27.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.27.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.27.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.27.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.28.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.28.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.28.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.28.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.29.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.29.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.29.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.29.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.3.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.3.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.3.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.3.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.30.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.30.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.30.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.30.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.31.att_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.31.attn_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.31.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.31.ff_proj.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.blocks.4.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.4.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.4.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.4.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.5.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.5.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.5.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.5.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.6.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.6.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.6.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.6.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.7.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.7.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.7.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.7.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.8.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.8.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.8.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.8.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.9.att_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.9.attn_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.9.ff_out.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.blocks.9.ff_proj.weight": "model-00001-of-00002.safetensors",
+        "model.transformer.ff_out.weight": "model-00002-of-00002.safetensors",
+        "model.transformer.wte.weight": "model-00001-of-00002.safetensors"
+    }
+}

modeling_olmo.py

@@ -0,0 +1,3 @@
+from hf_olmo import OLMoForCausalLM
+
+

pytorch_model-00001-of-00002.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c01e8e5db3174103ebbf4435dfb9c746e6fb2eb890966429e4a9e182ee3c277
+size 9945777322

pytorch_model-00002-of-00002.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3ae639bb44aca919f0c8f0966824d2f4dfe2eca5719eba0d5d70a02ed8cfbec
+size 3830460110

pytorch_model.bin.index.json

@@ -0,0 +1,137 @@
+{
+  "metadata": {
+    "total_size": 13776191488
+  },
+  "weight_map": {
+    "model.transformer.blocks.0.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.0.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.0.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.0.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.1.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.1.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.1.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.1.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.10.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.10.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.10.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.10.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.11.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.11.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.11.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.11.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.12.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.12.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.12.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.12.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.13.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.13.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.13.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.13.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.14.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.14.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.14.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.14.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.15.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.15.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.15.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.15.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.16.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.16.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.16.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.16.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.17.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.17.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.17.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.17.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.18.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.18.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.18.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.18.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.19.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.19.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.19.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.19.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.2.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.2.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.2.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.2.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.20.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.20.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.20.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.20.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.21.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.21.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.21.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.21.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.22.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.22.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.22.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.22.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.23.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.23.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.23.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.23.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.24.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.24.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.24.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.24.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.25.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.25.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.25.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.25.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.26.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.26.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.26.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.26.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.27.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.27.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.27.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.27.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.28.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.28.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.28.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.28.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.29.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.29.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.29.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.29.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.3.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.3.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.3.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.3.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.30.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.30.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.30.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.30.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.31.att_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.31.attn_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.31.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.31.ff_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.blocks.4.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.4.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.4.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.4.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.5.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.5.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.5.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.5.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.6.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.6.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.6.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.6.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.7.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.7.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.7.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.7.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.8.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.8.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.8.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.8.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.9.att_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.9.attn_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.9.ff_out.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.blocks.9.ff_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "model.transformer.ff_out.weight": "pytorch_model-00002-of-00002.bin",
+    "model.transformer.wte.weight": "pytorch_model-00001-of-00002.bin"
+  }
+}

special_tokens_map.json

@@ -0,0 +1,5 @@
+{
+  "eos_token": "<|endoftext|>",
+  "pad_token": "<|padding|>"
+}
+

tokenization_olmo_fast.py

@@ -0,0 +1,2 @@
+from hf_olmo.tokenization_olmo_fast import OLMoTokenizerFast
+

tokenizer_config.json

@@ -0,0 +1,243 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "|||IP_ADDRESS|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "1": {
+      "content": "<|padding|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50254": {
+      "content": "                        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50255": {
+      "content": "                       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50256": {
+      "content": "                      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50257": {
+      "content": "                     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50258": {
+      "content": "                    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50259": {
+      "content": "                   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50260": {
+      "content": "                  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50261": {
+      "content": "                 ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50262": {
+      "content": "                ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50263": {
+      "content": "               ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50264": {
+      "content": "              ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50265": {
+      "content": "             ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50266": {
+      "content": "            ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50267": {
+      "content": "           ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50268": {
+      "content": "          ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50269": {
+      "content": "         ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50270": {
+      "content": "        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50271": {
+      "content": "       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50272": {
+      "content": "      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50273": {
+      "content": "     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50274": {
+      "content": "    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50275": {
+      "content": "   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50276": {
+      "content": "  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50277": {
+      "content": "|||EMAIL_ADDRESS|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50278": {
+      "content": "|||PHONE_NUMBER|||",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50279": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|endoftext|>",
+  "max_length": null,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<|padding|>",
+  "tokenizer_class": "OLMoTokenizer",
+  "truncation": "right",
+  "auto_map": {
+    "AutoConfig": "configuration_olmo.OLMoConfig",
+    "AutoTokenizer": [
+      "tokenization_olmo_fast.OLMoTokenizerFast",
+      "tokenization_olmo_fast.OLMoTokenizerFast"
+    ]
+  },
+  "chat_template": "{{ eos_token }}{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n'  + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}"
+}