a  d$@srdZddlmZddlmZddlmZddlmZe e Z ddd d d d Z Gd ddeZ GdddeZdS)z T5 model configuration)Mapping)PretrainedConfig)OnnxSeq2SeqConfigWithPast)loggingz8https://huggingface.co/t5-small/resolve/main/config.jsonz7https://huggingface.co/t5-base/resolve/main/config.jsonz8https://huggingface.co/t5-large/resolve/main/config.jsonz5https://huggingface.co/t5-3b/resolve/main/config.jsonz6https://huggingface.co/t5-11b/resolve/main/config.json)zt5-smallzt5-basezt5-largezt5-3bzt5-11bcs8eZdZdZdZdgZddddZdfdd ZZS)T5ConfigaI This is the configuration class to store the configuration of a [`T5Model`] or a [`TFT5Model`]. It is used to instantiate a T5 model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the T5 [t5-small](https://huggingface.co/t5-small) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Arguments: vocab_size (`int`, *optional*, defaults to 32128): Vocabulary size of the T5 model. Defines the number of different tokens that can be represented by the `inputs_ids` passed when calling [`T5Model`] or [`TFT5Model`]. d_model (`int`, *optional*, defaults to 512): Size of the encoder layers and the pooler layer. d_kv (`int`, *optional*, defaults to 64): Size of the key, query, value projections per attention head. The `inner_dim` of the projection layer will be defined as `num_heads * d_kv`. d_ff (`int`, *optional*, defaults to 2048): Size of the intermediate feed forward layer in each `T5Block`. num_layers (`int`, *optional*, defaults to 6): Number of hidden layers in the Transformer encoder. num_decoder_layers (`int`, *optional*): Number of hidden layers in the Transformer decoder. Will use the same value as `num_layers` if not set. num_heads (`int`, *optional*, defaults to 8): Number of attention heads for each attention layer in the Transformer encoder. relative_attention_num_buckets (`int`, *optional*, defaults to 32): The number of buckets to use for each attention layer. relative_attention_max_distance (`int`, *optional*, defaults to 128): The maximum distance of the longer sequences for the bucket separation. dropout_rate (`float`, *optional*, defaults to 0.1): The ratio for all dropout layers. layer_norm_eps (`float`, *optional*, defaults to 1e-6): The epsilon used by the layer normalization layers. initializer_factor (`float`, *optional*, defaults to 1): A factor for initializing all weight matrices (should be kept to 1, used internally for initialization testing). feed_forward_proj (`string`, *optional*, defaults to `"relu"`): Type of feed forward layer to be used. Should be one of `"relu"` or `"gated-gelu"`. T5v1.1 uses the `"gated-gelu"` feed forward projection. Original T5 uses `"relu"`. use_cache (`bool`, *optional*, defaults to `True`): Whether or not the model should return the last key/values attentions (not used by all models). Zt5Zpast_key_valuesd_model num_heads num_layers)Z hidden_sizeZnum_attention_headsZnum_hidden_layers}@N 皙?ư>?reluTrc s||_||_||_||_||_|dur*|n|j|_||_||_| |_| |_ | |_ | |_ | |_ ||_ |j d}|d|_|ddk|_t|dkr|ddkst|dkrtd| d| d krd |_tjf|||d |dS) N-rZgatedrz`feed_forward_proj`: z is not a valid activation function of the dense layer.Please make sure `feed_forward_proj` is of the format `gated-{ACT_FN}` or `{ACT_FN}`, e.g. 'gated-gelu' or 'relu'z gated-geluZgelu_new) pad_token_id eos_token_idis_encoder_decoder) vocab_sizerd_kvd_ffr num_decoder_layersr relative_attention_num_bucketsrelative_attention_max_distance dropout_ratelayer_norm_epsiloninitializer_factorfeed_forward_proj use_cachesplitZ dense_act_fnZ is_gated_actlen ValueErrorsuper__init__)selfrrrr r r!r r"r#r$r%r&r'rr(rrkwargsZact_info __class__p/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/models/t5/configuration_t5.pyr-Rs>  $ zT5Config.__init__)r r r rrNrrrrrrrTTrr) __name__ __module__ __qualname____doc__Z model_typeZkeys_to_ignore_at_inferenceZ attribute_mapr- __classcell__r2r2r0r3r"s,+ rc@s@eZdZeeeeeeffdddZeedddZdS) T5OnnxConfig)returncCsxddddddd}|jrDd|dd<ddi|d <dd d|d <ndd d|d <dd d|d <|jrt|j|d d|S)NbatchZencoder_sequence)rr)Z input_idsattention_maskz past_encoder_sequence + sequencer<rrZdecoder_input_idsz past_decoder_sequence + sequenceZdecoder_attention_maskZdecoder_sequenceinputs) direction)Zuse_pastZfill_with_past_key_values_)r.Z common_inputsr2r2r3r=s  zT5OnnxConfig.inputscCsdS)N r2)r.r2r2r3default_onnx_opsetszT5OnnxConfig.default_onnx_opsetN) r4r5r6propertyrstrintr=r@r2r2r2r3r9s r9N)r7typingrZconfiguration_utilsrZonnxrutilsrZ get_loggerr4loggerZ T5_PRETRAINED_CONFIG_ARCHIVE_MAPrr9r2r2r2r3s      m