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