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

AxolotlInputConfig

utils.schemas.config.AxolotlInputConfig()

Wrapper of all config options.

EBFTConfig

utils.schemas.config.EBFTConfig()

Configuration for Energy-Based Fine-Tuning (EBFT)