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 |
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
Functions
| Name | Description |
|---|---|
| attn_impl_base | Strip the @revision / :kernel_name suffix so org/name@v2 still matches above. |
attn_impl_base
utils.schemas.enums.attn_impl_base(attn_implementation)Strip the @revision / :kernel_name suffix so org/name@v2 still matches above.