Update README.md

README.md

@@ -1,3 +1,20 @@
----
-license: mit
----
+---
+license: mit
+datasets:
+- openai/gsm8k
+base_model:
+- meta-llama/Meta-Llama-3-8B
+library_name: transformers
+---
+
+#### MuToR: Multi-Token prediction with Registers
+Arxiv: [https://arxiv.org/abs/2505.10518](https://arxiv.org/abs/2505.10518)
+
+**TL;DR**: **MuToR** is a simple, plug-and-play approach for multi-token prediction. 
+It leverages dummy register tokens to predict multiple targets in the future, enriching the supervisory signal and improving performance across diverse settings and modalities. The register tokens are discarded on inference, leaving generation speed unchanged.
+
+---
+
+#### Model Description
+This model is a finetuned version of **Llama 3 8B**. It was finetuned using the MuToR method for 5 epochs on the GSM8K training split.
+Please refer to our [code](https://github.com/nasosger/MuToR) for guidelines on how to use the models to reproduce our results.