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**:
|