turingmachine commited on
Commit
ff594dd
1 Parent(s): 5bd1290

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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="turingmachine/hupd-distilroberta-base")
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("turingmachine/hupd-distilroberta-base")
66
- model = AutoModelForMaskedLM.from_pretrained("turingmachine/hupd-distilroberta-base").to(device)
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