Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 53 |
).cuda().eval()
|
| 54 |
|
| 55 |
# A partially-masked input text string
|
| 56 |
-
text = "En søt lundefugl flyr over de<mask>norske fjorder."
|
| 57 |
input_ids = tokenizer(text, return_tensors='pt').input_ids.cuda()
|
| 58 |
|
| 59 |
# An empty attention mask allows uncontrained bidirectional attention
|
|
|
|
| 53 |
).cuda().eval()
|
| 54 |
|
| 55 |
# A partially-masked input text string
|
| 56 |
+
text = "En søt lundefugl flyr over de<mask>norske fjorder."
|
| 57 |
input_ids = tokenizer(text, return_tensors='pt').input_ids.cuda()
|
| 58 |
|
| 59 |
# An empty attention mask allows uncontrained bidirectional attention
|