core.builders.causal

core.builders.causal

Builder for causal trainers

Classes

Name	Description
HFCausalTrainerBuilder	Build the HuggingFace training args/trainer for causal models and reward modeling

HFCausalTrainerBuilder

core.builders.causal.HFCausalTrainerBuilder(
    cfg,
    model,
    tokenizer,
    processor=None,
)

Build the HuggingFace training args/trainer for causal models and reward modeling using TRL.