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