如何在消费级显卡运行 HiDream-I1 全家桶

HiDream-I1 是一个全新的开源图像生成基础模型，拥有 17B 个参数，可在数秒内实现最先进的图像生成质量。

感谢 https://github.com/hykilpikonna/HiDream-I1-nf4

我们提供完整版和精简版模型。它们的参数大小相同，因此运行时所需的 GPU 内存量也相同。不过，由于减少了推理步骤，蒸馏模型的运行速度更快。

名称	最小VRAM	步数	HuggingFace
HiDream-I1-Full	16 GB	50	🤗 Original / NF4
HiDream-I1-Dev	16 GB	28	🤗 Original / NF4
HiDream-I1-Fast	16 GB	16	🤗 Original / NF4

硬件要求

• GPU 架构: NVIDIA >= Ampere (e.g. A100, H100, A40, RTX 3090, RTX 4090)
• GPU 显存: >= 16 GB
• CPU 显存: >= 16 GB

快速上手

pip install hdi1 --no-build-isolation

[!NOTE]
建议为该软件包启动一个新的 python 环境，以避免依赖冲突。
为此，您可以使用 conda create -n hdi1 python=3.12 然后 conda activate hdi1 。
； 或者您可以使用 python3 -m venv venv 然后 source venv/bin/activate Linux 上或 venv\Scripts\activate在 Windows 上。

命令行界面

然后运行模块生成图像：

python -m hdi1 "A cat holding a sign that says 'hello world'"

# 或指定模型
python -m hdi1 "A cat holding a sign that says 'hello world'" -m fast

网络控制面板

我们还提供了用于交互式图像生成的网络控制面板。您可以通过运行

python -m hdi1.web