Hub Python Library documentation

安装

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

安装

在开始之前,您需要通过安装适当的软件包来设置您的环境

huggingface_hub 在 Python 3.8 或更高版本上进行了测试,可以保证在这些版本上正常运行。如果您使用的是 Python 3.7 或更低版本,可能会出现兼容性问题

使用 pip 安装

我们建议将huggingface_hub安装在虚拟环境中. 如果你不熟悉 Python虚拟环境,可以看看这个指南.

虚拟环境可以更容易地管理不同的项目,避免依赖项之间的兼容性问题

首先在你的项目目录中创建一个虚拟环境,请运行以下代码:

python -m venv .env

在Linux和macOS上,请运行以下代码激活虚拟环境:

source .env/bin/activate

在 Windows 上,请运行以下代码激活虚拟环境:

.env/Scripts/activate

现在您可以从PyPi注册表安装 huggingface_hub

pip install --upgrade huggingface_hub

完成后,检查安装是否正常工作

安装可选依赖项

huggingface_hub的某些依赖项是 可选 的,因为它们不是运行huggingface_hub的核心功能所必需的.但是,如果没有安装可选依赖项, huggingface_hub 的某些功能可能会无法使用

您可以通过pip安装可选依赖项,请运行以下代码:

# 安装 TensorFlow 特定功能的依赖项
# /!\ 注意:这不等同于 `pip install tensorflow`
pip install 'huggingface_hub[tensorflow]'

# 安装 TensorFlow 特定功能和 CLI 特定功能的依赖项
pip install 'huggingface_hub[cli,torch]'

这里列出了 huggingface_hub 的可选依赖项:

  • cli:为 huggingface_hub 提供更方便的命令行界面

  • fastai,torch, tensorflow: 运行框架特定功能所需的依赖项

  • dev:用于为库做贡献的依赖项。包括 testing(用于运行测试)、typing(用于运行类型检查器)和 quality(用于运行 linter)

从源代码安装

在某些情况下,直接从源代码安装huggingface_hub会更有趣。因为您可以使用最新的主版本main而非最新的稳定版本

main版本更有利于跟进平台的最新开发进度,例如,在最近一次官方发布之后和最新的官方发布之前所修复的某个错误

但是,这意味着main版本可能不总是稳定的。我们会尽力让其正常运行,大多数问题通常会在几小时或一天内解决。如果您遇到问题,请创建一个 Issue ,以便我们可以更快地解决!

pip install git+https://github.com/huggingface/huggingface_hub  # 使用pip从GitHub仓库安装Hugging Face Hub库

从源代码安装时,您还可以指定特定的分支。如果您想测试尚未合并的新功能或新错误修复,这很有用

pip install git+https://github.com/huggingface/huggingface_hub@my-feature-branch  # 使用pip从指定的GitHub分支(my-feature-branch)安装Hugging Face Hub库

完成安装后,请检查安装是否正常工作

可编辑安装

从源代码安装允许您设置可编辑安装.如果您计划为huggingface_hub做出贡献并需要测试代码更改,这是一个更高级的安装方式。您需要在本机上克隆一个huggingface_hub的本地副本

# 第一,使用以下命令克隆代码库
git clone https://github.com/huggingface/huggingface_hub.git

# 然后,使用以下命令启动虚拟环境
cd huggingface_hub
pip install -e .

这些命令将你克隆存储库的文件夹与你的 Python 库路径链接起来。Python 现在将除了正常的库路径之外,还会在你克隆到的文件夹中查找。例如,如果你的 Python 包通常安装在./.venv/lib/python3.11/site-packages/中,Python 还会搜索你克隆的文件夹./huggingface_hub/

通过 conda 安装

如果你更熟悉它,你可以使用conda-forge channel渠道来安装 huggingface_hub

请运行以下代码:

conda install -c conda-forge huggingface_hub

完成安装后,请检查安装是否正常工作

验证安装

安装完成后,通过运行以下命令检查huggingface_hub是否正常工作:

python -c "from huggingface_hub import model_info; print(model_info('gpt2'))"

这个命令将从 Hub 获取有关 gpt2 模型的信息。

输出应如下所示:

Model Name: gpt2  模型名称
Tags: ['pytorch', 'tf', 'jax', 'tflite', 'rust', 'safetensors', 'gpt2', 'text-generation', 'en', 'doi:10.57967/hf/0039', 'transformers', 'exbert', 'license:mit', 'has_space']  标签
Task: text-generation  任务:文本生成

Windows局限性

为了实现让每个人都能使用机器学习的目标,我们构建了 huggingface_hub库,使其成为一个跨平台的库,尤其可以在 Unix 和 Windows 系统上正常工作。但是,在某些情况下,huggingface_hub在Windows上运行时会有一些限制。以下是一些已知问题的完整列表。如果您遇到任何未记录的问题,请打开 Github上的issue.让我们知道

  • huggingface_hub的缓存系统依赖于符号链接来高效地缓存从Hub下载的文件。在Windows上,您必须激活开发者模式或以管理员身份运行您的脚本才能启用符号链接。如果它们没有被激活,缓存系统仍然可以工作,但效率较低。有关更多详细信息,请阅读缓存限制部分。

  • Hub上的文件路径可能包含特殊字符(例如:path/to?/my/file)。Windows对特殊字符更加严格,这使得在Windows上下载这些文件变得不可能。希望这是罕见的情况。如果您认为这是一个错误,请联系存储库所有者或我们,以找出解决方案。

后记

一旦您在机器上正确安装了huggingface_hub,您可能需要配置环境变量或者查看我们的指南之一以开始使用。

< > Update on GitHub