eli4s
/

Bert-L12-h384-A6

Inference Endpoints

Model card Files Files and versions Community

eli4s commited on Jul 22, 2021

Commit

408c74f

·

1 Parent(s): 3da2e1f

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -26,6 +26,7 @@ import torch
 sentence = "Let's have a [MASK]."
 encoded_inputs = tokenizer([sentence], padding='longest')
 input_ids = torch.tensor(encoded_inputs['input_ids'])
 attention_mask = torch.tensor(encoded_inputs['attention_mask'])

 sentence = "Let's have a [MASK]."
+model.eval()
 encoded_inputs = tokenizer([sentence], padding='longest')
 input_ids = torch.tensor(encoded_inputs['input_ids'])
 attention_mask = torch.tensor(encoded_inputs['attention_mask'])