from modal_training_gym.common.models.base import HFModelConfigurationModelConfig for models hosted on HuggingFace.
Inherits from: ModelConfig
Constructor
Section titled “Constructor”HFModelConfiguration(**kwargs)| Parameter | Type | Default | Description |
|---|
Attributes
Section titled “Attributes”| Attribute | Type | Default | Description |
|---|---|---|---|
model_name | str | "" | |
model_path | str | None | None | |
architecture | ModelArchitecture | None | None | |
response_parser | Optional[Callable[[str], ParsedResponse]] | None |
Methods
Section titled “Methods”download(self) -> 'None'
Section titled “download(self) -> '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.