Spaces:

sidphbot
/

Researcher

Build error

sidphbot commited on May 29, 2022

Commit

0e63d4b

1 Parent(s): d1763ec

t5 led switch

Files changed (1) hide show

src/Surveyor.py CHANGED Viewed

@@ -135,7 +135,7 @@ class Surveyor:
                 self.ledtokenizer = LEDTokenizer.from_pretrained(ledmodel_name)
                 self.ledmodel = LEDForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
             else:
-                self.ledtokenizer = T5Tokenizer.from_pretrained(ledmodel_name)
                 self.ledmodel = T5ForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
             self.ledmodel.eval()
             if not no_save_models:

                 self.ledtokenizer = LEDTokenizer.from_pretrained(ledmodel_name)
                 self.ledmodel = LEDForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
             else:
+                self.ledtokenizer = AutoTokenizer.from_pretrained(ledmodel_name)
                 self.ledmodel = T5ForConditionalGeneration.from_pretrained(ledmodel_name).to(self.torch_device)
             self.ledmodel.eval()
             if not no_save_models: