Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,10 @@ from transformers import AutoTokenizer, AutoModelWithLMHead
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt2-swahili")
|
| 19 |
|
| 20 |
model = AutoModelWithLMHead.from_pretrained("flax-community/gpt2-swahili")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
```
|
| 22 |
|
| 23 |
#### **Training Data**:
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("flax-community/gpt2-swahili")
|
| 19 |
|
| 20 |
model = AutoModelWithLMHead.from_pretrained("flax-community/gpt2-swahili")
|
| 21 |
+
|
| 22 |
+
print(round((model.num_parameters())/(1000*1000)),"Million Parameters")
|
| 23 |
+
|
| 24 |
+
124 Million Parameters
|
| 25 |
```
|
| 26 |
|
| 27 |
#### **Training Data**:
|