utils.schemas.enums

utils.schemas.enums

Enums for Axolotl input config

Classes

Name	Description
ChatTemplate	Chat templates configuration subset
CustomSupportedOptimizers	Custom supported optimizers
RLType	RL trainer type configuration subset
RingAttnFunc	Enum class for supported ring-flash-attn implementations
TorchIntDType	Torch integer data types - getattr guards against torch < 2.6 which does not support int4

ChatTemplate

utils.schemas.enums.ChatTemplate()

Chat templates configuration subset

CustomSupportedOptimizers

utils.schemas.enums.CustomSupportedOptimizers()

Custom supported optimizers

RLType

utils.schemas.enums.RLType()

RL trainer type configuration subset

RingAttnFunc

utils.schemas.enums.RingAttnFunc()

Enum class for supported ring-flash-attn implementations

TorchIntDType

utils.schemas.enums.TorchIntDType()

Torch integer data types - getattr guards against torch < 2.6 which does not support int4