ModelConfiguration

ModelConfiguration

from modal_training_gym.common.models.base import ModelConfiguration

Base class for model identity and weight-download logic.

Constructor

ModelConfiguration(**kwargs)

Parameter	Type	Default	Description

Attributes

Attribute	Type	Default	Description
model_name	str	""	HuggingFace repo ID or other model identifier. Default "".
model_path	`str	None`	None
architecture	`ModelArchitecture	None`	None
training	`ModelTrainingConfig	None`	None

Methods

download_model(self) -> 'None'

Download or materialize weights into the model volume.

Related Tutorials

• Shared concepts: config containers, framework factories, volume layout, running the pipeline
• Custom HuggingFace model (SmolLM2-135M) LoRA SFT — inline ModelConfiguration subclass, no catalog entry

Source: modal_training_gym/common/models/base.py