Skip to content
GitHub
View on GitHub

ModelConfig

Base class for model identity and weight-download logic.

from modal_training_gym.common.models.base import ModelConfig

Base class for model identity and weight-download logic.

ModelConfig(**kwargs)
ParameterTypeDefaultDescription
AttributeTypeDefaultDescription
model_namestr""
model_pathstr | NoneNone
architectureModelArchitecture | NoneNone
response_parserOptional[Callable[[str], ParsedResponse]]None

Download or materialize weights into the model volume.

parse_response(self, text: 'str') -> 'ParsedResponse'

Section titled “parse_response(self, text: 'str') -> 'ParsedResponse'”

Parse raw model output into structured content.

Source: modal_training_gym/common/models/base.py