o  h8@sddlZddlmZmZddlmZddlmZmZddl m Z ddl m Z ddl mZeeZeee jGd d d e ZdS) N) dataclassfield)Path)OptionalUnion)GenerationConfig)TrainingArguments)add_start_docstringscseZdZUdZedddidZeed<edddidZeed<ed dd idZ e e ed <ed dd idZ e e ed <ed ddidZ e eeeefed<fddZZS)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. Fhelpz%Whether to use SortishSampler or not.)defaultmetadatasortish_samplerzFWhether to use generate to calculate generative metrics (ROUGE, BLEU).predict_with_generateNzThe `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_lengthzThe `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_configcs6t}|D]\}}t|tr|||<q |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. )superto_dictitems isinstancer)selfdkv __class__v/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/transformers/training_args_seq2seq.pyrPs   z Seq2SeqTrainingArguments.to_dict)__name__ __module__ __qualname____doc__rrbool__annotations__rrrintrrrstrrrr __classcell__rrrrr s0   r )logging dataclassesrrpathlibrtypingrrgeneration.configuration_utilsr training_argsr utilsr getLoggerr loggerr#r rrrrs