Huggingface Hub 的官方 Python 用戶端。
英語|德語 | हिंदी | 한국어 | 中文(簡體)
文件:https://hf.co/docs/huggingface_hub
原始碼:https://github.com/huggingface/huggingface_hub
huggingface_hub
庫可讓您與 Hugging Face Hub 進行交互,這是一個為創作者和合作者實現開源機器學習民主化的平台。發現適合您的專案的預訓練模型和資料集,或使用 Hub 上託管的數千個機器學習應用程式。您還可以建立並與社群分享您自己的模型、資料集和演示。 huggingface_hub
函式庫提供了一種使用 Python 完成所有這些操作的簡單方法。
使用 pip 安裝huggingface_hub
套件:
pip install huggingface_hub
如果您願意,也可以使用 conda 安裝它。
為了在預設情況下保持套件最小化, huggingface_hub
附帶了對某些用例有用的可選依賴項。例如,如果您想獲得完整的推理體驗,請執行:
pip install huggingface_hub[inference]
若要了解更多安裝和可選依賴項,請查看安裝指南。
下載單一文件
from huggingface_hub import hf_hub_download
hf_hub_download ( repo_id = "tiiuae/falcon-7b-instruct" , filename = "config.json" )
或整個儲存庫
from huggingface_hub import snapshot_download
snapshot_download ( "stabilityai/stable-diffusion-2-1" )
文件將下載到本機快取資料夾。本指南中有更多詳細資訊。
Hugging Face Hub 使用令牌來驗證應用程式(請參閱文件)。若要登入您的計算機,請執行以下 CLI:
huggingface-cli login
# or using an environment variable
huggingface-cli login --token $HUGGINGFACE_TOKEN
from huggingface_hub import create_repo
create_repo ( repo_id = "super-cool-model" )
上傳單一文件
from huggingface_hub import upload_file
upload_file (
path_or_fileobj = "/home/lysandre/dummy-test/README.md" ,
path_in_repo = "README.md" ,
repo_id = "lysandre/test-model" ,
)
或者整個資料夾
from huggingface_hub import upload_folder
upload_folder (
folder_path = "/path/to/local/space" ,
repo_id = "username/my-cool-space" ,
repo_type = "space" ,
)
詳情請參閱上傳指南。
我們正在與超酷的開源 ML 庫合作,提供免費的模型託管和版本控制。您可以在此處找到現有的整合。
優點是:
如果您想整合您的庫,請隨時打開一個問題來開始討論。我們編寫了一份分步指南,其中 ❤️ 展示瞭如何進行此整合。
歡迎每個人做出貢獻,我們重視每個人的貢獻。程式碼並不是幫助社區的唯一方法。回答問題、幫助他人、接觸和改進文件對於社群來說非常有價值。我們編寫了一份貢獻指南來總結如何開始為此儲存庫做出貢獻。