Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

6c14705

•

1 Parent(s): 3073c57

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -6,6 +6,7 @@ from dagshub.pytorch_lightning import DAGsHubLogger
 from transformers import (
     AdamW,
     T5ForConditionalGeneration,
 )
 from torch.utils.data import Dataset, DataLoader
 import pytorch_lightning as pl

 from transformers import (
     AdamW,
     T5ForConditionalGeneration,
+    T5TokenizerFast as T5Tokenizer, MT5Tokenizer, MT5ForConditionalGeneration, ByT5Tokenizer,
 )
 from torch.utils.data import Dataset, DataLoader
 import pytorch_lightning as pl