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
