开源协议: MIT
基础模型:

• HiDream-ai/HiDream-I1-Dev
  任务类型: 文生图
  库名称: diffusers
  基础模型关系: 量化版本

HiDream-I1 4比特量化模型

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

原始仓库请访问此处。

HiDream-I1是拥有170亿参数的开源图像生成基础模型，能在数秒内生成达到业界顶尖水平的图像。

模型版本

我们提供完整版和蒸馏版模型。参数规模相同，因此运行时所需显存相同。但蒸馏版通过减少推理步数实现更快生成速度。

硬件要求

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

快速开始

执行安装命令：

pip install hdi1 --no-build-isolation

[!注意]
建议新建Python环境以避免依赖冲突：

• 使用conda: conda create -n hdi1 python=3.12 → conda activate hdi1
• 或使用venv: 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模型文件，需先签署Llama模型许可协议并通过huggingface-cli login登录。

网页控制台

启动交互式图像生成界面：

python -m hdi1.web  

许可协议

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