ltg
/

norbert3-small

Norwegian Bokmål

Norwegian Nynorsk

Model card Files Files and versions Community

davda54 commited on Jun 29, 2023

Commit

57f0e08

·

1 Parent(s): e49af6f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ model = AutoModelForMaskedLM.from_pretrained("ltg/norbert3-small", trust_remote_
 mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
 input_text = tokenizer("Nå ønsker de seg en[MASK] bolig.", return_tensors="pt")
-output_p = bert(**input_text)
 output_text = torch.where(input_text.input_ids == mask_id, output_p.logits.argmax(-1), input_text.input_ids)
 # should output: '[CLS] Nå ønsker de seg en ny bolig.[SEP]'

 mask_id = tokenizer.convert_tokens_to_ids("[MASK]")
 input_text = tokenizer("Nå ønsker de seg en[MASK] bolig.", return_tensors="pt")
+output_p = model(**input_text)
 output_text = torch.where(input_text.input_ids == mask_id, output_p.logits.argmax(-1), input_text.input_ids)
 # should output: '[CLS] Nå ønsker de seg en ny bolig.[SEP]'