o
    ZhN                     @   sz   d Z ddlmZ ddlmZmZmZ ddlmZ e	e
ZdZG dd	 d	eZG d
d deZG dd deZg dZdS )zTensorflow mT5 model.   )logging   )TFT5EncoderModelTFT5ForConditionalGeneration	TFT5Model   )	MT5ConfigZT5Configc                   @      e Zd ZdZdZeZdS )
TFMT5Modela   
    This class overrides [`TFT5Model`]. Please check the superclass for the appropriate documentation alongside usage
    examples.

    Examples:

    ```python
    >>> from transformers import TFMT5Model, AutoTokenizer

    >>> model = TFMT5Model.from_pretrained("google/mt5-small")
    >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
    >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien."
    >>> summary = "Weiter Verhandlung in Syrien."
    >>> inputs = tokenizer(article, return_tensors="tf")
    >>> labels = tokenizer(text_target=summary, return_tensors="tf")

    >>> outputs = model(input_ids=inputs["input_ids"], decoder_input_ids=labels["input_ids"])
    >>> hidden_states = outputs.last_hidden_state
    ```mt5N__name__
__module____qualname____doc__Z
model_typer   Zconfig_class r   r   V/var/www/auris/lib/python3.10/site-packages/transformers/models/mt5/modeling_tf_mt5.pyr
      s    r
   c                   @   r	   )TFMT5ForConditionalGenerationa  
    This class overrides [`TFT5ForConditionalGeneration`]. Please check the superclass for the appropriate
    documentation alongside usage examples.

    Examples:

    ```python
    >>> from transformers import TFMT5ForConditionalGeneration, AutoTokenizer

    >>> model = TFMT5ForConditionalGeneration.from_pretrained("google/mt5-small")
    >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
    >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien."
    >>> summary = "Weiter Verhandlung in Syrien."
    >>> inputs = tokenizer(article, text_target=summary, return_tensors="tf")

    >>> outputs = model(**inputs)
    >>> loss = outputs.loss
    ```r   Nr   r   r   r   r   r   4   s    r   c                   @   r	   )TFMT5EncoderModelan  
    This class overrides [`TFT5EncoderModel`]. Please check the superclass for the appropriate documentation alongside
    usage examples.

    Examples:

    ```python
    >>> from transformers import TFMT5EncoderModel, AutoTokenizer

    >>> model = TFMT5EncoderModel.from_pretrained("google/mt5-small")
    >>> tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
    >>> article = "UN Offizier sagt, dass weiter verhandelt werden muss in Syrien."
    >>> input_ids = tokenizer(article, return_tensors="tf").input_ids
    >>> outputs = model(input_ids)
    >>> hidden_state = outputs.last_hidden_state
    ```r   Nr   r   r   r   r   r   L   s    r   )r   r   r
   N)r   utilsr   Zt5.modeling_tf_t5r   r   r   Zconfiguration_mt5r   Z
get_loggerr   loggerZ_CONFIG_FOR_DOCr
   r   r   __all__r   r   r   r   <module>   s   
