seidel
/

plsum-base-ptt5

text2text-generation

text-generation-inference

Model card Files Files and versions

seidel commited on Apr 28, 2022

Commit

6d38e8a

·

1 Parent(s): e8acd89

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ Project [github](https://github.com/aseidelo/wiki_generator/tree/cdd38918c207020
   tokenizer = T5TokenizerFast.from_pretrained("seidel/plsum-base-ptt5")
   model = T5ForConditionalGeneration.from_pretrained("seidel/plsum-base-ptt5", use_cache=False)
-  x = tokenizer([input_text], padding="max_length", max_length=max_length, return_tensors="pt", truncation=True)
   y = model.generate(**x)
   print(tokenizer.batch_decode(y, skip_special_tokens=True))

   tokenizer = T5TokenizerFast.from_pretrained("seidel/plsum-base-ptt5")
   model = T5ForConditionalGeneration.from_pretrained("seidel/plsum-base-ptt5", use_cache=False)
+  x = tokenizer([input_text], padding="max_length", max_length=512, return_tensors="pt", truncation=True)
   y = model.generate(**x)
   print(tokenizer.batch_decode(y, skip_special_tokens=True))