Generative Representational Instruction Tuning (GRIT) • 63 items • Updated • 9
Model Summary
GritLM is a generative representational instruction tuned language model. It unifies text representation (embedding) and text generation into a single model achieving state-of-the-art performance on both types of tasks.
- Repository: ContextualAI/gritlm
- Paper: https://arxiv.org/abs/2402.09906
- Logs: https://wandb.ai/muennighoff/gritlm/runs/0uui712t/overview
- Script: https://github.com/ContextualAI/gritlm/blob/main/scripts/training/train_gritlm_7b.sh
| Model | Description |
|---|---|
| GritLM 7B | Mistral 7B finetuned using GRIT |
| GritLM 8x7B | Mixtral 8x7B finetuned using GRIT |
Use
The model usage is documented here.
Citation
@misc{muennighoff2024generative,
title={Generative Representational Instruction Tuning},
author={Niklas Muennighoff and Hongjin Su and Liang Wang and Nan Yang and Furu Wei and Tao Yu and Amanpreet Singh and Douwe Kiela},
year={2024},
eprint={2402.09906},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
- Downloads last month
- 10,075
Safetensors
Model size
7B params
Tensor type
BF16
·
Model tree for GritLM/GritLM-7B
Base model
mistralai/Mistral-7B-v0.1Dataset used to train GritLM/GritLM-7B
Spaces using GritLM/GritLM-7B 49
Collection including GritLM/GritLM-7B
Paper for GritLM/GritLM-7B
Evaluation results
- mteb/arguana leaderboard
- ArguAna Default Test View evaluation results source Obtained using MTEB v1.12.7563.17 *
- ArguAna View evaluation results source Obtained using MTEB v1.12.7563.17 *
- accuracy on MTEB AmazonCounterfactualClassification (en)test set self-reported81.179
- ap on MTEB AmazonCounterfactualClassification (en)test set self-reported46.263
- f1 on MTEB AmazonCounterfactualClassification (en)test set self-reported75.446
- accuracy on MTEB AmazonPolarityClassificationtest set self-reported96.516
- ap on MTEB AmazonPolarityClassificationtest set self-reported94.791
