Downloads · 30 days
172
5% of all-time downloads
gpt2-swahili
flax-community/gpt2-swahili
gpt2-swahili is a text generation model from flax-community. Use it when you need the model to write or continue text. It is set up for transformers.
This model was trained using HuggingFace's Flax framework and is part of the JAX/Flax Community Week organized by HuggingFace. All training was done on a TPUv3-8 VM sponsored by the Google Cloud team.
All-time downloads
3.4K
Public
Repo size
1.5 GB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 50%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 25, 2021
- SHA
- 8ca5ef61
Try a prompt
Datasets
Model card
From the Hugging Face model README
GPT2 in Swahili
This model was trained using HuggingFace's Flax framework and is part of the JAX/Flax Community Week organized by HuggingFace. All training was done on a TPUv3-8 VM sponsored by the Google Cloud team.
How to use
from transformers import AutoTokenizer, AutoModelWithLMHead
tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt2-swahili")
model = AutoModelWithLMHead.from_pretrained("flax-community/gpt2-swahili")
print(round((model.num_parameters())/(1000*1000)),"Million Parameters")
124 Million Parameters
Training Data:
This model was trained on Swahili Safi
More Details:
For more details and Demo please check HF Swahili Space
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Languages
- sw
- Created
- Mar 2, 2022
- Updated
- Jul 25, 2021