Update README.md
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ To use the pre-trained model for masked language modeling, use the following sni
|
|
| 15 |
```python
|
| 16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
| 17 |
|
| 18 |
-
# See the `
|
| 19 |
tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
|
| 20 |
model = AutoModelForMaskedLM.from_pretrained('s-sahoo/duo-distilled')
|
| 21 |
```
|
|
|
|
| 15 |
```python
|
| 16 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
| 17 |
|
| 18 |
+
# See the `DUO` collection page on the hub for list of available models.
|
| 19 |
tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
|
| 20 |
model = AutoModelForMaskedLM.from_pretrained('s-sahoo/duo-distilled')
|
| 21 |
```
|