RoGPT2-base / README.md
MihaiAlexandru1606's picture
init
4f84de3

Model card for RoGPT2-base


language: - ro

RoGPT2: Romanian GPT2 for text generation

All models are available:

How to use

~~from transformers import AutoTokenizer, TFAutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-base')
model = TFAutoModelWithLMHead.from_pretrained('readerbench/RoGPT2-base')
inputs = tokenizer.encode("Este o zi de vara", return_tensors='tf')
text = model.generate(inputs, max_length=1024,  no_repeat_ngram_size=2)
print(tokenizer.decode(text[0]))~~