Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 14, 2021

Commit

87281da

•

1 Parent(s): d50b787

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -54,3 +54,6 @@ class DataModule(Dataset):
             max_length=self.source_max_token_len,
             padding="max_length",
             truncation=True,

             max_length=self.source_max_token_len,
             padding="max_length",
             truncation=True,
+            return_attention_mask=True,
+            add_special_tokens=True,
+            return_tensors="pt",