turingmachine
commited on
Commit
•
ff594dd
1
Parent(s):
5bd1290
Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ You can use this model directly with a pipeline for masked language modeling:
|
|
25 |
|
26 |
```python
|
27 |
from transformers import pipeline
|
28 |
-
model = pipeline(task="fill-mask", model="
|
29 |
model("Improved <mask> for playing a game of thumb wrestling.")
|
30 |
```
|
31 |
|
@@ -62,8 +62,8 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
|
|
62 |
# cuda/cpu
|
63 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
64 |
|
65 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
66 |
-
model = AutoModelForMaskedLM.from_pretrained("
|
67 |
|
68 |
TEXT = "Improved <mask> for playing a game of thumb wrestling."
|
69 |
|
|
|
25 |
|
26 |
```python
|
27 |
from transformers import pipeline
|
28 |
+
model = pipeline(task="fill-mask", model="hupd/hupd-distilroberta-base")
|
29 |
model("Improved <mask> for playing a game of thumb wrestling.")
|
30 |
```
|
31 |
|
|
|
62 |
# cuda/cpu
|
63 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
64 |
|
65 |
+
tokenizer = AutoTokenizer.from_pretrained("hupd/hupd-distilroberta-base")
|
66 |
+
model = AutoModelForMaskedLM.from_pretrained("hupd/hupd-distilroberta-base").to(device)
|
67 |
|
68 |
TEXT = "Improved <mask> for playing a game of thumb wrestling."
|
69 |
|