btqkhai commited on
Commit
1d93277
·
verified ·
1 Parent(s): 0cfc904

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -19,7 +19,7 @@ model = BertForMaskedLM.from_pretrained('btqkhai/SinoNomBERT')
19
 
20
  text = '大 [MASK] 百 官 其 𢮿 花 供 饌 皆 用 新 禮'
21
 
22
- inputs = tokenizer(text, return_tensors="pt", add_special_tokens=True)
23
  mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
24
  # Ground Truth: 宴
25
  logits = model(**inputs).logits
 
19
 
20
  text = '大 [MASK] 百 官 其 𢮿 花 供 饌 皆 用 新 禮'
21
 
22
+ inputs = tokenizer(text, return_tensors="pt")
23
  mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
24
  # Ground Truth: 宴
25
  logits = model(**inputs).logits