license: mit base_model:

• HiDream-ai/HiDream-I1-Fast pipeline_tag: text-to-image library_name: diffusers base_model_relation: quantized

HiDream-I1 4比特量化模型

本仓库是HiDream-I1的4比特量化版本分支，量化后完整模型可在16GB显存以下运行。

原始仓库地址请访问此处。

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

模型版本

我们提供完整版和蒸馏版模型。参数规模相同，因此运行时需要相同的GPU显存。但蒸馏版模型因推理步骤减少而速度更快。

硬件要求

• GPU架构: NVIDIA ≥安培架构 (如A100/H100/A40/RTX 3090/RTX 4090)
• 显存容量: ≥16GB
• 内存容量: ≥16GB

快速开始

执行以下命令安装：

pip install hdi1 --no-build-isolation

[!注意] 建议新建Python环境以避免依赖冲突
可通过conda create -n hdi1 python=3.12创建环境后conda activate hdi1激活
或使用python3 -m venv venv创建虚拟环境，Linux系统执行source venv/bin/activate，Windows执行venv\Scripts\activate

命令行界面

安装后可通过模块生成图像：

python -m hdi1 "举着'hello world'标牌的猫咪"

# 可指定模型版本
python -m hdi1 "举着'hello world'标牌的猫咪" -m fast

[!注意] 推理脚本会自动下载meta-llama/Llama-3.1-8B-Instruct模型文件。使用前需在HuggingFace账户同意Llama模型许可协议，并通过huggingface-cli login登录。

网页控制台

我们还提供交互式图像生成网页控制台，启动命令：

python -m hdi1.web

许可协议

本仓库代码及HiDream-I1模型均采用MIT许可协议。