HuggingFaceModelInfo
Information about a HuggingFace model to store.
Attributes
| Attribute | Type | Description |
|---|---|---|
| repo | str | The HuggingFace repository ID (e.g., 'meta-llama/Llama-2-7b-hf'). |
| artifact_name | `str | None` = None |
| architecture | `str | None` = None |
| task | str = "auto" | Model task (e.g., 'generate', 'classify', 'embed'). |
| modality | tuple[str, ...] = ("text",) | Modalities supported by the model (e.g., 'text', 'image'). |
| serial_format | `str | None` = None |
| model_type | `str | None` = None |
| short_description | `str | None` = None |
| shard_config | `ShardConfig | None` = None |