开源协议: mit
基础模型:

• HiDream-ai/HiDream-I1-Full
  任务类型: 文生图
  库名称: 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 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开源协议。