| license: mit | |
| language: | |
| - ar | |
| ## Checkpoints | |
| ### Pre-Trained Models | |
| Model | Pre-train Dataset | Model | Tokenizer | | |
| | --- | --- | --- | --- | | |
| | ArTST v2 base | Dialects | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/pretrain_checkpoint.pt) | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/tokenizer_artstv2.model) | |
| ### Finetuned Models | |
| Model | FInetune Dataset | Model | Tokenizer | | |
| | --- | --- | --- | --- | | |
| | ArTST v2 ASR | MGB2 | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/ASR_MGB2_best.pt_hf.pt) | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/tokenizer_artstv2.model) | | |
| | ArTST v2 ASR | QASR | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/ASR_QASR_best.pt_hf.pt) | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/tokenizer_artstv2.model) | | |
| | ArTST v2 ASR | MGB2 - Dialects | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/ASR_Dialects_MGB2_best.pt_hf.pt) | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/tokenizer_artstv2.model) | | |
| | ArTST v2 ASR | MGB2 - Dialects | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/ASR_Dialects_QASR_best.pt_hf.pt) | [Hugging Face](https://huggingface.co/MBZUAI/ArTSTv2/blob/main/tokenizer_artstv2.model) | | |
| # Acknowledgements | |
| ArTST is built on [SpeechT5](https://arxiv.org/abs/2110.07205) Architecture. If you use any of ArTST models, please cite | |
| ``` | |
| @inproceedings{toyin2023artst, | |
| title={ArTST: Arabic Text and Speech Transformer}, | |
| author={Toyin, Hawau and Djanibekov, Amirbek and Kulkarni, Ajinkya and Aldarmaki, Hanan}, | |
| booktitle={Proceedings of ArabicNLP 2023}, | |
| pages={41--51}, | |
| year={2023} | |
| } | |
| ``` |