utils.schemas.model

utils.schemas.model

Pydantic models for model input / output, etc. configuration

Classes

Name Description
ModelInputConfig Model configuration subset
ModelOutputConfig model save configuration subset
SpecialTokensConfig Special tokens configuration subset

ModelInputConfig

utils.schemas.model.ModelInputConfig()

Model configuration subset

ModelOutputConfig

utils.schemas.model.ModelOutputConfig()

model save configuration subset

SpecialTokensConfig

utils.schemas.model.SpecialTokensConfig()

Special tokens configuration subset