Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

下載模型

整合庫

如果 Hub 上的模型與支援的庫相關聯,只需幾行程式碼即可載入模型。有關訪問模型的資訊,您可以單擊模型頁面上的“在*庫*中使用”按鈕,以檢視如何操作。例如,distilbert/distilgpt2 顯示瞭如何使用下面的 🤗 Transformers。

使用 Hugging Face 客戶端庫

您可以使用 huggingface_hub 庫來建立、刪除、更新和檢索儲存庫中的資訊。例如,要從命令列下載 HuggingFaceH4/zephyr-7b-beta 模型,請執行:

hf download HuggingFaceH4/zephyr-7b-beta

有關更多資訊,請參閱 CLI 下載文件

您還可以將其整合到您自己的庫中。例如,您可以用幾行程式碼快速載入 Scikit-learn 模型。

from huggingface_hub import hf_hub_download
import joblib

REPO_ID = "YOUR_REPO_ID"
FILENAME = "sklearn_model.joblib"

model = joblib.load(
    hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
)

使用 Git

由於模型中心上的所有模型都是 Git 儲存庫,您可以透過執行以下命令在本地克隆模型:

git lfs install
git clone git@hf.co:<MODEL ID> # example: git clone git@hf.co:bigscience/bloom

如果您對特定模型儲存庫具有寫入許可權,您也將能夠提交和推送對模型的修訂。

將您的 SSH 公鑰新增到您的使用者設定以推送更改和/或訪問私有儲存庫。

更快的下載速度

如果您在具有高頻寬的機器上執行,您可以使用 hf_transfer 提高下載速度,這是一個基於 Rust 的庫,旨在加速與 Hub 的檔案傳輸。

pip install "huggingface_hub[hf_transfer]"
HF_HUB_ENABLE_HF_TRANSFER=1 hf download ...

hf_transfer 是一個高階使用者工具!它經過測試且已投入生產,但它缺少使用者友好的功能,如高階錯誤處理或代理。有關更多詳細資訊,請參閱此指南

< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.