LanguageModelDPOLossConfig

Module: fast_llm.layers.language_model.loss.config

Variant of: LanguageModelLossConfig — select with type: dpo

Inherits from: LanguageModelLossConfig

Fields

beta — core

Type: float    Default: 1.0

Beta parameter for DPO loss (controls strength of preference optimization).

weight — core

Type: float    Default: 1.0

Weight for this loss in the total loss computation.

logits_scale_factor — feature

Type: float    Default: 1.0

Extra logits scale factor applied for this loss only, stacked on top of the model's logits_scale_factor.

reference_model — feature

Type: str    Default: (required)

Name of the reference model to use for dpo.