utils.schemas.config
utils.schemas.config
Module with Pydantic models for configuration.
Classes
| Name | Description |
|---|---|
| AxolotlConfigWCapabilities | Wrapper to valdiate GPU capabilities with the configured options |
| AxolotlInputConfig | Wrapper of all config options. |
| EBFTConfig | Configuration for Energy-Based Fine-Tuning (EBFT) |
AxolotlConfigWCapabilities
utils.schemas.config.AxolotlConfigWCapabilities()Wrapper to valdiate GPU capabilities with the configured options
Methods
| Name | Description |
|---|---|
| check_fp8_attention_preflight | fp8 attention requires SM90+ and torch >= 2.11 (torchao >= 0.17 is pinned). |
check_fp8_attention_preflight
utils.schemas.config.AxolotlConfigWCapabilities.check_fp8_attention_preflight()fp8 attention requires SM90+ and torch >= 2.11 (torchao >= 0.17 is pinned).
AxolotlInputConfig
utils.schemas.config.AxolotlInputConfig()Wrapper of all config options.
Methods
| Name | Description |
|---|---|
| normalize_attn_implementation | Map legacy boolean attention flags to canonical attn_implementation, warn, then strip. |
| validate_attn_implementation | Accept canonical names and hub-kernel paths; reject short-form aliases. |
normalize_attn_implementation
utils.schemas.config.AxolotlInputConfig.normalize_attn_implementation(data)Map legacy boolean attention flags to canonical attn_implementation, warn, then strip.
validate_attn_implementation
utils.schemas.config.AxolotlInputConfig.validate_attn_implementation(value)Accept canonical names and hub-kernel paths; reject short-form aliases.
EBFTConfig
utils.schemas.config.EBFTConfig()Configuration for Energy-Based Fine-Tuning (EBFT)