Model Card
An Arabic LLM derived from Google's mT5 multi-lingual model
Model Details
Model Description
This is a smaller version of the google/mt5-base model with only Arabic and some English embeddings left.
The original model has 582M parameters, with 384M of them being input and output embeddings. After shrinking the sentencepiece vocabulary from 250K to 30K (top 10K English and top 20K Arabic tokens) the number of model parameters reduced to 244M parameters, and model size reduced from 2.2GB to 0.9GB - 42% of the original one.
The creation of this model was inspired from David Dales'article "How to adapt a multilingual T5 model for a single language" in which mT5 was compressed to support Russian and English languages along with the source code.
- Developed by: Moustafa Banbouk
- Model type: Unsupervised LLM
- Language(s) (NLP): Arabic, English
- License: MIT
Downstream Uses
Quesion Answering, Summarization, Classification ...
- Downloads last month
- 7
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("ArabicNLP/mT5-base_ar") model = AutoModelForSeq2SeqLM.from_pretrained("ArabicNLP/mT5-base_ar")