csebuetnlp
/

mT5_multilingual_XLSum

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

AbhishekRaghuvanshi commited on Aug 16, 2023

Commit

e2791ce

•

1 Parent(s): 2437a52

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -130,12 +130,12 @@ input_ids = tokenizer(
     return_tensors="pt",
     padding="max_length",
     truncation=True,
-    max_length=512
 )["input_ids"]
 output_ids = model.generate(
     input_ids=input_ids,
-    max_length=84,
     no_repeat_ngram_size=2,
     num_beams=4
 )[0]

     return_tensors="pt",
     padding="max_length",
     truncation=True,
+    max_length=5120
 )["input_ids"]
 output_ids = model.generate(
     input_ids=input_ids,
+    max_length=840,
     no_repeat_ngram_size=2,
     num_beams=4
 )[0]