readerbench
/

RoGPT2-large

Text Generation

text-generation-inference

Model card Files Files and versions

ReaderBench commited on Jul 20, 2021

Commit

e60b3cd

·

1 Parent(s): 7100155

update readme

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -30,7 +30,7 @@ print(tokenizer.decode(text[0]))
 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
-model = AutoModelWithLMHead.from_pretrained('readerbench/RoGPT2-large')
 inputs = tokenizer.encode("Este o zi de vara", return_tensors='pt')
 text = model.generate(inputs, max_length=1024,  no_repeat_ngram_size=2)
 print(tokenizer.decode(text[0]))

 from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
+model = AutoModelForCausalLM.from_pretrained('readerbench/RoGPT2-large')
 inputs = tokenizer.encode("Este o zi de vara", return_tensors='pt')
 text = model.generate(inputs, max_length=1024,  no_repeat_ngram_size=2)
 print(tokenizer.decode(text[0]))