o
    ¹ÈÏiN  ã                   @   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é   )Ú	MT5ConfigÚ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__Ú
model_typer   Úconfig_class© r   r   ú[/home/ubuntu/.local/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   Út5.modeling_tf_t5r   r   r   Úconfiguration_mt5r   Ú
get_loggerr   ÚloggerÚ_CONFIG_FOR_DOCr   r   r   Ú__all__r   r   r   r   Ú<module>   s   
