Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

ae6c335

1 Parent(s): 272671c

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -359,6 +359,7 @@ class Summarization:
             self.model = T5ForConditionalGeneration.from_pretrained(
                 f"{model_dir}", return_dict=True
             )
         if use_gpu:
             if torch.cuda.is_available():

             self.model = T5ForConditionalGeneration.from_pretrained(
                 f"{model_dir}", return_dict=True
             )
+        elif model_type == "mt5":
         if use_gpu:
             if torch.cuda.is_available():