U &c@s:ddlZddlZddlmZeeZGdddeZdS)N)PretrainedConfigcsBeZdZdZdZfddZeeeedddZdd Z Z S) EncoderDecoderConfiga :class:`~transformers.EncoderDecoderConfig` is the configuration class to store the configuration of a `EncoderDecoderModel`. It is used to instantiate an Encoder Decoder model according to the specified arguments, defining the encoder and decoder configs. Configuration objects inherit from :class:`~transformers.PretrainedConfig` and can be used to control the model outputs. See the documentation for :class:`~transformers.PretrainedConfig` for more information. Args: kwargs (`optional`): Remaining dictionary of keyword arguments. Notably: encoder (:class:`PretrainedConfig`, optional, defaults to `None`): An instance of a configuration object that defines the encoder config. encoder (:class:`PretrainedConfig`, optional, defaults to `None`): An instance of a configuration object that defines the decoder config. Example:: from transformers import BertConfig, EncoderDecoderConfig, EncoderDecoderModel # Initializing a BERT bert-base-uncased style configuration config_encoder = BertConfig() config_decoder = BertConfig() config = EncoderDecoderConfig.from_encoder_decoder_configs(config_encoder, config_decoder) # Initializing a Bert2Bert model from the bert-base-uncased style configurations model = EncoderDecoderModel(config=config) # Accessing the model configuration config_encoder = model.config.encoder config_decoder = model.config.decoder Zencoder_decoderc stjf|d|krd|ks&td|d}|d}|d}|d}ddlm}|j|f||_|j|f||_d|_ dS)Nencoderdecoderzs     zEncoderDecoderConfig.__init__)rrreturncCs|||dS)a Instantiate a :class:`~transformers.EncoderDecoderConfig` (or a derived class) from a pre-trained encoder model configuration and decoder model configuration. Returns: :class:`EncoderDecoderConfig`: An instance of a configuration object )rr)to_dict)clsrrrrrfrom_encoder_decoder_configsNs z1EncoderDecoderConfig.from_encoder_decoder_configscCs8t|j}|j|d<|j|d<|jj|d<|S)z Serializes this instance to a Python dictionary. Override the default `to_dict()` from `PretrainedConfig`. Returns: :obj:`Dict[str, any]`: Dictionary of all the attributes that make up this configuration instance, rrr)copydeepcopy__dict__rrrrr)r outputrrrrZs   zEncoderDecoderConfig.to_dict) __name__ __module__ __qualname____doc__rr classmethodrrr __classcell__rrrrrs!  r)rloggingZconfiguration_utilsr getLoggerrloggerrrrrrs