Skip to content
GitHub
View on GitHub

HFModelConfiguration

API reference for HFModelConfiguration

from modal_training_gym.common.models.base import HFModelConfiguration

ModelConfiguration for models hosted on HuggingFace.

Inherits from: ModelConfiguration

HFModelConfiguration(**kwargs)
ParameterTypeDefaultDescription
AttributeTypeDefaultDescription
model_namestr""HuggingFace repo ID or other model identifier. Default "".
model_path`strNone`None
architecture`ModelArchitectureNone`None
training`ModelTrainingConfigNone`None

Download or materialize weights into the model volume.

Source: modal_training_gym/common/models/base.py