基于 HiDream-I1 模型的图像生成服务，提供 REST API 和 Web 界面。
- 使用 HiDream-I1 模型进行高质量图像生成
- 提供 REST API 接口
- 提供用户友好的 Web 界面
- 支持自定义生成参数
- 支持 GPU 加速
- 自动服务监控和重启
- 自动模型下载管理
- NVIDIA GPU (推荐)
- Docker 和 Docker Compose
- NVIDIA Container Toolkit
- 至少 16GB 内存
- 至少 20GB 磁盘空间
- Hugging Face 账号和访问令牌
- 克隆仓库：
git clone https://github.com/fengqiaozhu/hidream.git
cd hidream- 设置环境变量：
export HF_TOKEN=your_huggingface_token- 启动服务：
docker-compose up -d服务首次启动时会自动检查并下载所需的模型文件。下载完成后服务会自动启动。
- 访问服务：
- Web 界面: http://localhost:7860
- API 文档: http://localhost:8000/docs
模型文件会被保存在 ./models 目录中。服务启动时会自动检查以下模型：
- HiDream-I1-Full
- Meta-Llama-3.1-8B-Instruct
如果模型不存在，系统会自动下载。你也可以手动下载模型并放置在对应目录中：
./models/
├── HiDream-I1-Full/
└── Meta-Llama-3.1-8B-Instruct/GET /generate
参数：
prompt: 图像描述文本steps: 推理步数 (1-100, 默认: 50)guidance_scale: 引导比例 (1.0-20.0, 默认: 7.5)height: 图像高度 (256-1024, 默认: 512)width: 图像宽度 (256-1024, 默认: 512)seed: 随机种子 (可选)
- 安装依赖：
pip install -r requirements.txt- 下载模型：
python scripts/download_models.py --token $HF_TOKEN- 启动服务：
python app.py # API 服务
python web.py # Web 界面docker-compose build.
├── Dockerfile # Docker 构建配置
├── app.py # API 服务
├── web.py # Web 界面
├── docker-compose.yml # Docker 编排配置
├── supervisord.conf # 进程管理配置
└── scripts/
├── start.sh # 启动脚本（包含模型检查）
└── download_models.py # 模型下载脚本
MIT License
欢迎提交 Issue 和 Pull Request！
如有问题，请通过 GitHub Issues 联系我们。