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(*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
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.