AKI.SafetensorМодели и датасеты
← Все мини-уроки

SDK · 5 МИН

Скачайте модель через Python

Подключите huggingface_hub к AKI.Safetensor и получите файлы.

Что понадобится

Python, действующий аккаунт с доступом и токен для чтения. Публичную карточку можно смотреть без входа, а скачивание требует предоставленного доступа.

1. Установите клиент

python -m pip install huggingface_hub

2. Подготовьте окружение

В кабинете откройте «Подключение» и скопируйте значение HF_ENDPOINT. Сохраните его в переменной окружения HF_ENDPOINT. Создайте токен «Только чтение» и передайте его в AI_VAULT_TOKEN через настройки окружения своей системы. Не вставляйте секрет в исходный код.

В AI_VAULT_REPO_ID укажите идентификатор выбранной модели, например owner/name из её карточки.

3. Скачайте файлы

import os
from huggingface_hub import snapshot_download

path = snapshot_download(
    repo_id=os.environ["AI_VAULT_REPO_ID"],
    endpoint=os.environ["HF_ENDPOINT"],
    token=os.environ["AI_VAULT_TOKEN"],
    local_dir="./downloaded-model",
)
print(path)

Для конкретного релиза добавьте параметр revision с существующей веткой, тегом или коммитом репозитория. Проверьте доступное место перед передачей больших файлов.

Если доступ запрещён

Проверьте адрес HF_ENDPOINT, срок доступа организации, срок токена и права на репозиторий. Повторный вход в браузере сам по себе не обновляет токен в Python.

Результат

Файлы находятся в downloaded-model. Для запуска используйте инструкции автора модели и собственную вычислительную среду.