From 97b787fb4e59168ca1c9c329884fe24a5292d001 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Tue, 19 Jan 2021 09:56:25 -0500 Subject: [PATCH] Fix old Seq2SeqTrainer (#9675) --- examples/seq2seq/seq2seq_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/seq2seq/seq2seq_trainer.py b/examples/seq2seq/seq2seq_trainer.py index 32a965551e..cba3e958e9 100644 --- a/examples/seq2seq/seq2seq_trainer.py +++ b/examples/seq2seq/seq2seq_trainer.py @@ -60,7 +60,7 @@ class Seq2SeqTrainer(Trainer): assert isinstance( self.model, PreTrainedModel ), f"If no `config` is passed the model to be trained has to be of type `PreTrainedModel`, but is {self.model.__class__}" - self.config = self._actual_model(self.model).config + self.config = self.model.config else: self.config = config