Flexolmo 7x7B 1T RT

🚀 FlexOlmo

FlexOlmo is a new type of language model that enables a novel paradigm of data collaboration, allowing data owners to contribute to open - source language model development without relinquishing data control.

🚀 Quick Start

To use FlexOlmo, you need to install transformers from this source and run the following code:

from transformers import Olmoe2ForCausalLM, AutoTokenizer
import torch

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

MODEL_NAME = "allenai/FlexOlmo-7x7B-1T"
model = Olmoe2ForCausalLM.from_pretrained(MODEL_NAME).to(DEVICE)
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
inputs = tokenizer("Bitcoin is", return_tensors="pt")
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
out = model.generate(**inputs, max_length=64)
print(tokenizer.decode(out[0]))

✨ Features

FlexOlmo is a new kind of LM that unlocks a new paradigm of data collaboration. With FlexOlmo, data owners can contribute to the development of open language models without giving up control of their data. There is no need to share raw data directly, and data contributors can decide when their data is active in the model, deactivate it at any time, and receive attributions whenever it's used for inference.

📚 Documentation

Model Summary

FlexOlmo - 7x7B - 1T (without router training) is a Mixture - of - Experts with 33B total parameters, combining independently trained experts on public - mix, news, math, code, academic texts, creative writing, and Reddit data. The public - mix expert is trained on 1T tokens of public data while the other experts are branched from the public - mix expert and trained on 50B tokens of their respective data.

This information and more can also be found:

• Paper: https://allenai.org/papers/flexolmo
• Code: https://github.com/allenai/FlexOlmo
• Blog: https://allenai.org/blog/flexolmo
• Data and corresponding models: | Corpus | Public | Math | News | Academic | Code | Creative Writing | Reddit | | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | | Model | [Flex - public - 7B - 1T](https://huggingface.co/allenai/Flex - public - 7B - 1T) | [Flex - math - 2x7B - 1T](https://huggingface.co/allenai/Flex - math - 2x7B - 1T) | [Flex - news - 2x7B - 1T](https://huggingface.co/allenai/Flex - news - 2x7B - 1T) | [Flex - pes2o - 2x7B - 1T](https://huggingface.co/allenai/Flex - pes2o - 2x7B - 1T) | [Flex - code - 2x7B - 1T](https://huggingface.co/allenai/Flex - code - 2x7B - 1T) | [Flex - creative - 2x7B - 1T](https://huggingface.co/allenai/Flex - creative - 2x7B - 1T) | [Flex - reddit - 2x7B - 1T](https://huggingface.co/allenai/Flex - reddit - 2x7B - 1T) |

Evaluation Snapshot

• The evaluation of the individual model refers to the dense model, not the 2x7B MoE model.

📄 License

This project is licensed under the Apache - 2.0 license.

📚 Citation

@misc{flexolmo,
      title={FlexOlmo: Open Language Models for Flexible Data Use}, 
      author={Weijia Shi and Akshita Bhagia and Kevin Farhat and Niklas Muennighoff and Pete Walsh and Jacob Morrison and Dustin Schwenk and Shayne Longpre and Jake Poznanski and Allyson Ettinger and Daogao Liu and Margaret Li and Mike Lewis and Wen - tau Yih and Dirk Groeneveld and Luca Soldaini and Kyle Lo and Noah A. Smith and Luke Zettlemoyer and Pang Wei Koh and Hannaneh Hajishirzi and Ali Farhadi and Sewon Min},
      year={2025},
      eprint={2507.00000},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://allenai.org/papers/flexolmo}, 
}