Update README.md
Browse files
README.md
CHANGED
@@ -13,7 +13,7 @@ large batch size of 0.5M tokens. A larger 762 million parameter model can also b
|
|
13 |
#### How to use
|
14 |
|
15 |
```python
|
16 |
-
from transformers import
|
17 |
model = MegatronBertModel.from_pretrained("mmukh/SOBertBase")
|
18 |
tokenizer = BertTokenizerFast.from_pretrained("mmukh/SOBertBase")
|
19 |
|
|
|
13 |
#### How to use
|
14 |
|
15 |
```python
|
16 |
+
from transformers import MegatronBertModel,BertTokenizerFast
|
17 |
model = MegatronBertModel.from_pretrained("mmukh/SOBertBase")
|
18 |
tokenizer = BertTokenizerFast.from_pretrained("mmukh/SOBertBase")
|
19 |
|