DiffuGen - Advanced Local Image Generator with MCP Integration

基本能力

产品定位

DiffuGen 是一个嵌入开发环境的 AI 艺术工作室，旨在为开发者提供高效的图像生成工具，支持多种 AI 模型，并可通过 MCP 协议与 IDE 集成。

核心功能

1. 多模型支持：支持 Flux Schnell、Flux Dev、SDXL、SD3 和 SD1.5 等多种模型。
2. MCP 集成：与支持 MCP 的 IDE（如 Cursor、Windsurf 等）无缝集成。
3. OpenAPI 支持：提供 REST API 接口，支持直接 HTTP 访问图像生成功能。
4. 参数控制：支持调整图像尺寸、采样步数、CFG 比例、种子值等参数。
5. GPU 加速：利用 CUDA 加速，提升图像生成速度。
6. 自然语言接口：通过简单的自然语言命令生成图像。

适用场景

1. 开发辅助：在开发过程中快速生成 UI 设计、图标或概念图。
2. 创意设计：生成艺术图像、插画或海报设计。
3. 原型设计：快速生成产品原型或场景概念图。
4. 教育与研究：用于 AI 模型的教学和研究。

工具列表

1. generate_stable_diffusion_image：使用 Stable Diffusion 模型生成图像。
2. generate_flux_image：使用 Flux 模型生成图像。

常见问题解答

1. 模型缺失或路径错误：确保模型文件已下载并放置在正确目录。
2. CUDA/GPU 问题：更新 NVIDIA 驱动或设置 CUDA_VISIBLE_DEVICES。
3. 图像质量问题：调整步数、CFG 比例或采样方法。
4. 文件权限错误：确保输出目录可写，脚本有执行权限。

使用教程

使用依赖

1. Python 3.8+：确保已安装 Python 3.8 或更高版本。
2. CUDA（可选）：如需 GPU 加速，需安装 CUDA 驱动。

安装教程

自动安装（推荐）

git clone https://github.com/CLOUDWERX-DEV/diffugen.git
cd DiffuGen
chmod +x diffugen.sh
chmod +x setup_diffugen.sh
./setup_diffugen.sh

手动安装

1. 克隆仓库：

git clone https://github.com/CLOUDWERX-DEV/diffugen.git
cd DiffuGen
git clone --recursive https://github.com/leejet/stable-diffusion.cpp

1. 构建 stable-diffusion.cpp：

cd stable-diffusion.cpp
mkdir -p build && cd build
cmake .. -DCMAKE_BUILD_TYPE=Release -DSD_CUDA=ON
make -j$(nproc)
cd ../..

1. 创建 Python 虚拟环境：

python3 -m venv diffugen_env
source diffugen_env/bin/activate
pip install -r requirements.txt

1. 下载模型文件（详见 README）。

调试方式

1. 启动服务器：

cd /path/to/diffugen
./diffugen.sh

1. 检查日志：确保看到 DiffuGen ready 提示。
2. 测试生成：

./diffugen.sh "A futuristic cityscape with flying cars"

许可证

该项目遵循 MIT 开源许可条款，请参阅 MIT 了解完整条款。