§
    ‚b¦i  ã                   óÆ   — d dl Z d dlmZmZ d dlmZ ddlmZ ddlm	Z	 ddl
mZ  e j        e¦  «        Ze ee	j        ¦  «         G d„ d	e	¦  «        ¦   «         ¦   «         ZdS )
é    N)Ú	dataclassÚfield)ÚPathé   )ÚGenerationConfig)ÚTrainingArguments)Úadd_start_docstringsc                   ó
  ‡ — e Zd ZU dZ edddi¬¦  «        Zeed<    edddi¬¦  «        Zeed<    ed	dd
i¬¦  «        Z	e
d	z  ed<    ed	ddi¬¦  «        Ze
d	z  ed<    ed	ddi¬¦  «        Zeez  ez  d	z  ed<   ˆ fd„Zˆ xZS )ÚSeq2SeqTrainingArgumentsa®  
    Args:
        predict_with_generate (`bool`, *optional*, defaults to `False`):
            Whether to use generate to calculate generative metrics (ROUGE, BLEU).
        generation_max_length (`int`, *optional*):
            The `max_length` to use on each evaluation loop when `predict_with_generate=True`. Will default to the
            `max_length` value of the model configuration.
        generation_num_beams (`int`, *optional*):
            The `num_beams` to use on each evaluation loop when `predict_with_generate=True`. Will default to the
            `num_beams` value of the model configuration.
        generation_config (`str` or `Path` or [`~generation.GenerationConfig`], *optional*):
            Allows to load a [`~generation.GenerationConfig`] from the `from_pretrained` method. This can be either:

            - a string, the *model id* of a pretrained model configuration hosted inside a model repo on
              huggingface.co.
            - a path to a *directory* containing a configuration file saved using the
              [`~GenerationConfig.save_pretrained`] method, e.g., `./my_model_directory/`.
            - a [`~generation.GenerationConfig`] object.
    FÚhelpz%Whether to use SortishSampler or not.)ÚdefaultÚmetadataÚsortish_samplerzFWhether to use generate to calculate generative metrics (ROUGE, BLEU).Úpredict_with_generateNz•The `max_length` to use on each evaluation loop when `predict_with_generate=True`. Will default to the `max_length` value of the model configuration.Úgeneration_max_lengthz“The `num_beams` to use on each evaluation loop when `predict_with_generate=True`. Will default to the `num_beams` value of the model configuration.Úgeneration_num_beamsz^Model id, file path or url pointing to a GenerationConfig json file, to use during prediction.Úgeneration_configc                 óÔ   •— t          ¦   «                              ¦   «         }|                     ¦   «         D ]1\  }}t          |t          ¦  «        r|                     ¦   «         ||<   Œ2|S )zØ
        Serializes this instance while replace `Enum` by their values and `GenerationConfig` by dictionaries (for JSON
        serialization support). It obfuscates the token values by removing their value.
        )ÚsuperÚto_dictÚitemsÚ
isinstancer   )ÚselfÚdÚkÚvÚ	__class__s       €ú]/root/projects/butler/venv/lib/python3.11/site-packages/transformers/training_args_seq2seq.pyr   z Seq2SeqTrainingArguments.to_dictO   s]   ø€ õ ‰GŒGOŠOÑÔˆØ—G’G‘I”Ið 	#ð 	#‰DˆAˆqÝ˜!Õ-Ñ.Ô.ð #Ø—y’y‘{”{!‘øØˆó    )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚboolÚ__annotations__r   r   Úintr   r   Ústrr   r   r   Ú__classcell__)r   s   @r   r   r      sd  ø€ € € € € € ðð ð( "˜E¨%¸6ÐCjÐ:kÐlÑlÔl€OTÐlÐlÑlØ"' %Ø Ð)qÐ rð#ñ #ô #Ð˜4ð ð ñ ð ).¨ØàðHð
ð)ñ )ô )Ð˜3 ™:ð ð ñ ð (- uØàðGð
ð(ñ (ô (Ð˜# ™*ð ð ñ ð ?D¸eØàÐtð
ð?ñ ?ô ?Ðs˜T‘zÐ$4Ñ4°tÑ;ð ð ñ ð
ð 
ð 
ð 
ð 
ð 
ð 
ð 
ð 
r   r   )ÚloggingÚdataclassesr   r   Úpathlibr   Úgeneration.configuration_utilsr   Útraining_argsr   Úutilsr	   Ú	getLoggerr    Úloggerr#   r   © r   r   ú<module>r2      sß   ðð €€€Ø (Ð (Ð (Ð (Ð (Ð (Ð (Ð (Ø Ð Ð Ð Ð Ð à <Ð <Ð <Ð <Ð <Ð <Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø 'Ð 'Ð 'Ð 'Ð 'Ð 'ð 
ˆÔ	˜8Ñ	$Ô	$€ð ØÐÐ'Ô/Ñ0Ô0ð<ð <ð <ð <ð <Ð0ñ <ô <ñ 1Ô0ñ „ð<ð <ð <r   