Skip to content

LoRAConfig

class mint.types.LoraConfig ( StrictBase )

Fields:

  • rank ( int ) – LoRA rank (dimension of low-rank matrices)
  • seed ( Optional[int] ) – Seed used for initialization of LoRA weights.

Useful if you need deterministic or reproducible initialization of weights. * train_unembed ( bool ) – Whether to add lora to the unembedding layer * train_mlp ( bool ) – Whether to add loras to the MLP layers (including MoE layers) * train_attn ( bool ) – Whether to add loras to the attention layers