fixed typo

2020-03-09 20:01:20 +01:00
parent a2c8e516c2
commit 374deef48d
2 changed files with 2 additions and 2 deletions
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -855,7 +855,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin):
            cur_len = 1

            # put model in generation mode if it has one
-            if hasattr(self.model, "generation_mode"):
+            if hasattr(self.model, "decoder") and hasattr(self.model.decoder, "generation_mode"):
                self.model.decoder.generation_mode = True
        else:
            encoder_inputs = None