ReaderBench
commited on
Commit
•
e60b3cd
1
Parent(s):
7100155
update readme
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ print(tokenizer.decode(text[0]))
|
|
30 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
31 |
|
32 |
tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
|
33 |
-
model =
|
34 |
inputs = tokenizer.encode("Este o zi de vara", return_tensors='pt')
|
35 |
text = model.generate(inputs, max_length=1024, no_repeat_ngram_size=2)
|
36 |
print(tokenizer.decode(text[0]))
|
|
|
30 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
31 |
|
32 |
tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-large')
|
33 |
+
model = AutoModelForCausalLM.from_pretrained('readerbench/RoGPT2-large')
|
34 |
inputs = tokenizer.encode("Este o zi de vara", return_tensors='pt')
|
35 |
text = model.generate(inputs, max_length=1024, no_repeat_ngram_size=2)
|
36 |
print(tokenizer.decode(text[0]))
|