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` |
| architecture | `str | 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` |
| model_type | `str | None` |
| short_description | `str | None` |
| shard_config | `ShardConfig | None` |
Constructor
Signature
def HuggingFaceModelInfo(
repo: str,
artifact_name: str | None = None,
architecture: str | None = None,
task: str = "auto",
modality: tuple[str, ...] = ("text",),
serial_format: str | None = None,
model_type: str | None = None,
short_description: str | None = None,
shard_config: ShardConfig | None = None
) - > null
Parameters
| Name | 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 |
Signature
def HuggingFaceModelInfo(
repo: str,
artifact_name: str | None = null,
architecture: str | None = null,
task: str = "auto",
modality: tuple[str, ...] = ("text",),
serial_format: str | None = null,
model_type: str | None = null,
short_description: str | None = null,
shard_config: ShardConfig | None = null
) - > null
Parameters
| Name | Type | Description |
|---|---|---|
| repo | str | The HuggingFace repository ID, such as 'meta-llama/Llama-2-7b-hf', used to identify the source model. |
| artifact_name | `str | None` = null |
| architecture | `str | None` = null |
| task | str = "auto" | The intended model task, such as 'generate', 'classify', or 'embed', used to determine inference behavior. |
| modality | tuple[str, ...] = ("text",) | A collection of supported data modalities, such as 'text' or 'image'. |
| serial_format | `str | None` = null |
| model_type | `str | None` = null |
| short_description | `str | None` = null |
| shard_config | `ShardConfig | None` = null |