junnyu / roformer_small_generator

Other530Updated Sep 22, 2021

Downloads

53

Likes

0

Last updated

Sep 22, 2021

License

mit

text = "Beijing is the capital of [MASK]." tokenizer = ElectraTokenizer.frompretrained("junnyu/roformersmallgenerator") ptmodel = RoFormerForMaskedLM.frompretrained( "junnyu/roformersmallgenerator") ptinputs = tokenizer(text, returntensors="pt") with torch.nograd(): ptoutputs …

Tasks & library

  • fill mask
  • transformers

Tags

  • transformers
  • pytorch
  • roformer
  • fill-mask
  • electra
  • masked-lm

Technical specifications

ParametersNot available
Librarytransformers
Languagesen
Licensemit