Diffusers documentation

安装

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.31.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

安装

在你正在使用的任意深度学习框架中安装 🤗 Diffusers 。

🤗 Diffusers已在Python 3.8+、PyTorch 1.7.0+和Flax上进行了测试。按照下面的安装说明,针对你正在使用的深度学习框架进行安装:

  • PyTorch installation instructions.
  • Flax installation instructions.

使用pip安装

你需要在虚拟环境中安装 🤗 Diffusers 。

如果你对 Python 虚拟环境不熟悉,可以看看这个教程.

在虚拟环境中,你可以轻松管理不同的项目,避免依赖项之间的兼容性问题。

首先,在你的项目目录下创建一个虚拟环境:

python -m venv .env

激活虚拟环境:

source .env/bin/activate

现在,你就可以安装 🤗 Diffusers了!使用下边这个命令:

PyTorch

pip install diffusers["torch"]

Flax

pip install diffusers["flax"]

从源代码安装

在从源代码安装 diffusers 之前,确保你已经安装了 torchaccelerate

torch的安装教程可以看 torch 文档.

安装 accelerate

pip install accelerate

从源码安装 🤗 Diffusers 需要使用以下命令:

pip install git+https://github.com/huggingface/diffusers

这个命令安装的是最新的 main版本,而不是最近的stable版。 main是一直和最新进展保持一致的。比如,上次发布的正式版中有bug,在main中可以看到这个bug被修复了,但是新的正式版此时尚未推出。 但是这也意味着 main版本不保证是稳定的。

我们努力保持main版本正常运行,大多数问题都能在几个小时或一天之内解决

如果你遇到了问题,可以提 Issue,这样我们就能更快修复问题了。

可修改安装

如果你想做以下两件事,那你可能需要一个可修改代码的安装方式:

  • 使用 main版本的源代码。
  • 为 🤗 Diffusers 贡献,需要测试代码中的变化。

使用以下命令克隆并安装 🤗 Diffusers:

git clone https://github.com/huggingface/diffusers.git
cd diffusers

PyTorch

pip install -e ".[torch]"

Flax

pip install -e ".[flax]"

这些命令将连接到你克隆的版本库和你的 Python 库路径。 现在,不只是在通常的库路径,Python 还会在你克隆的文件夹内寻找包。 例如,如果你的 Python 包通常安装在 ~/anaconda3/envs/main/lib/python3.10/Site-packages/,Python 也会搜索你克隆到的文件夹。~/diffusers/

如果你想继续使用这个库,你必须保留 diffusers 文件夹。

现在你可以用下面的命令轻松地将你克隆的 🤗 Diffusers 库更新到最新版本。

cd ~/diffusers/
git pull

你的Python环境将在下次运行时找到main版本的 🤗 Diffusers。

注意 Telemetry 日志

我们的库会在使用from_pretrained()请求期间收集 telemetry 信息。这些数据包括Diffusers和PyTorch/Flax的版本,请求的模型或管道类,以及预训练检查点的路径(如果它被托管在Hub上的话)。 这些使用数据有助于我们调试问题并确定新功能的开发优先级。 Telemetry 数据仅在从 HuggingFace Hub 中加载模型和管道时发送,而不会在本地使用期间收集。

我们知道,并不是每个人都想分享这些的信息,我们尊重您的隐私, 因此您可以通过在终端中设置 DISABLE_TELEMETRY 环境变量从而禁用 Telemetry 数据收集:

Linux/MacOS :

export DISABLE_TELEMETRY=YES

Windows :

set DISABLE_TELEMETRY=YES
< > Update on GitHub