fix encoder decoder kwargs (#7131)

2020-09-15 21:10:07 +02:00
parent 4c62c6021a
commit 85ffda96fc
1 changed files with 1 additions and 1 deletions
@@ -250,7 +250,7 @@ class EncoderDecoderModel(PreTrainedModel):
                    encoder_config.is_decoder = False
                    encoder_config.add_cross_attention = False

-                    kwargs_encoder["config"] = encoder_config
+                kwargs_encoder["config"] = encoder_config

            encoder = AutoModel.from_pretrained(encoder_pretrained_model_name_or_path, *model_args, **kwargs_encoder)