core.trainers.dpo.trainer

core.trainers.dpo.trainer

DPO trainer for axolotl

Classes

Name	Description
AxolotlDPOTrainer	Extend the base DPOTrainer for axolotl helpers

AxolotlDPOTrainer

core.trainers.dpo.trainer.AxolotlDPOTrainer(
    self,
    *args,
    dataset_tags=None,
    **kwargs,
)

Extend the base DPOTrainer for axolotl helpers

Methods

Name	Description
push_to_hub	Overwrite the push_to_hub method in order to force-add the tags when pushing the

push_to_hub

core.trainers.dpo.trainer.AxolotlDPOTrainer.push_to_hub(*args, **kwargs)

Overwrite the push_to_hub method in order to force-add the tags when pushing the model on the Hub. Please refer to ~transformers.Trainer.push_to_hub for more details.