Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

b9ee162

•

1 Parent(s): 0f1daf7

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -265,6 +265,7 @@ class Summarization:
             self.model = MT5ForConditionalGeneration.from_pretrained(
                 f"{model_name}", return_dict=True
             )
     def train(
             self,

             self.model = MT5ForConditionalGeneration.from_pretrained(
                 f"{model_name}", return_dict=True
             )
+            self.tokenizer = ByT5Tokenizer.from_pretrained(f"{model_name}")
     def train(
             self,