Downloads · 30 days
130
24% of all-time downloads
MuToR-gemma-2B-GSM8K-dmax_4_a_03
nasos10/MuToR-gemma-2B-GSM8K-dmax_4_a_03
MuToR-gemma-2B-GSM8K-dmax_4_a_03 is a text generation model from nasos10. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
All-time downloads
538
Public
Parameters
2.5B
5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors5 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- mit
- Model type
- gemma
- Access
- Public
- Created
- Jun 27, 2025
- Updated
- Jun 27, 2025
- SHA
- 867285a9
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
MuToR: Multi-Token prediction with Registers
Arxiv: 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 Gemma-2B. It was finetuned using the MuToR method for 5 epochs on the GSM8K training split. Please refer to our code for guidelines on how to use the models to reproduce our results.
- Task
- Text Generation
- Library
- transformers
- Type
- gemma
- License
- mit
- Created
- Jun 27, 2025
- Updated
- Jun 27, 2025