ComfyUI-ModelScope-API 是一个 ComfyUI 自定义节点扩展，允许用户通过调用 ModelScope API 进行图像生成和编辑。该扩展支持多种基础模型和 LoRA 模型，提供了灵活的配置选项和直观的 Web UI 界面。
本扩展提供了以下核心功能：
- 文生图：支持多种基础模型和 LoRA 模型，可生成高质量图像
- 图像编辑：支持对现有图像进行编辑，同样支持多种基础模型和 LoRA 模型
- API 密钥管理：方便管理和切换 ModelScope API 密钥
- 模型预设管理：提供内置模型预设，支持自定义模型预设
- LoRA 预设管理：提供内置 LoRA 模型预设，支持自定义 LoRA 模型预设
首先，确保您已安装 ComfyUI 及其基本依赖。然后，安装本扩展所需的依赖：
pip install -r requirements.txt
依赖列表：
- requests>=2.28.0
- torch>=1.13.0
- Pillow>=9.0.0
- numpy>=1.21.0
将本扩展文件夹复制到 ComfyUI 的 custom_nodes 目录下：
cp -r ComfyUI-ModelScope-API /path/to/ComfyUI/custom_nodes/
然后重启 ComfyUI，扩展将自动加载。
该节点用于通过 ModelScope API 生成图像，支持多种基础模型和 LoRA 模型。
输入参数：
- 提示词：描述您想要生成的图像内容
- API Token：ModelScope API 令牌
- 基础模型：选择要使用的基础模型
- LoRA模型1-6：可选的 LoRA 模型，格式为 "model_name|weight"（weight 可选，默认 1.0）
- 负面提示词：描述您不想在图像中出现的内容
- 宽度/高度：生成图像的尺寸
- 随机种子：控制生成结果的随机性
- 采样步数：控制生成过程的迭代次数
- 引导系数：控制提示词对生成结果的影响程度
- 批次大小：一次生成的图像数量
输出：
- 生成的图像
该节点用于通过 ModelScope API 编辑现有图像，支持多种基础模型和 LoRA 模型。
输入参数：
- 图像：要编辑的原始图像
- 编辑提示词：描述您想要如何编辑图像
- API Token：ModelScope API 令牌
- 基础模型：选择要使用的基础模型
- LoRA模型1-6：可选的 LoRA 模型，格式为 "model_name|weight"（weight 可选，默认 1.0）
- 负面提示词：描述您不想在编辑后图像中出现的内容
- 使用自定义尺寸：是否使用自定义宽度和高度
- 宽度/高度：编辑后图像的尺寸
- 随机种子：控制编辑结果的随机性
- 采样步数：控制编辑过程的迭代次数
- 引导系数：控制提示词对编辑结果的影响程度
输出：
- 编辑后的图像
该节点用于管理 ModelScope API 密钥，方便用户切换和保存密钥。
该节点用于管理模型预设，提供内置模型预设，支持自定义模型预设。
该节点用于管理 LoRA 模型预设，提供内置 LoRA 模型预设，支持自定义 LoRA 模型预设。
首先，您需要在 ModelScope 官网 注册账号并获取 API 令牌。
有两种方式配置 API 密钥：
在使用 "魔搭API-文生图节点" 或 "魔搭API-图像编辑节点" 时，直接在节点的 "API Token" 输入框中输入您的 API 密钥，系统会自动保存。
编辑 nodes/.modelscope_api_token 文件，将您的 API 密钥写入该文件。
- 在 ComfyUI 工作区中添加 "魔搭API-文生图节点"。
- 输入您的提示词。
- 选择要使用的基础模型。
- （可选）添加 LoRA 模型。
- 设置其他参数（宽度、高度、采样步数等）。
- 连接节点输出到其他节点或保存到文件。
- 执行工作流，生成图像。
- 在 ComfyUI 工作区中添加 "魔搭API-图像编辑节点"。
- 连接要编辑的图像到节点输入。
- 输入您的编辑提示词。
- 选择要使用的基础模型。
- （可选）添加 LoRA 模型。
- 设置其他参数。
- 连接节点输出到其他节点或保存到文件。
- 执行工作流，编辑图像。
该文件包含节点的默认配置，如默认提示词、默认模型、超时设置等。
默认配置：
{
"default_lora_model": "qiyuanai/TikTok_Xiaohongshu_career_line_beauty_v1",
"timeout": 1200,
"image_download_timeout": 60,
"default_prompt": "Career line,with prominent breasts,A very realistic style,high definition photography style,a young woman,long black hair,holding a badminton shuttlecock,standing,outdoors.",
"default_edit_prompt": "修改图片中的内容"
}
内置模型预设配置文件，包含支持的基础模型列表。
用户自定义模型预设配置文件，用户可以在这里添加自己的模型预设。
内置 LoRA 模型预设配置文件，包含支持的 LoRA 模型列表。
用户自定义 LoRA 模型预设配置文件，用户可以在这里添加自己的 LoRA 模型预设。
| 模型名称 | 模型标识 |
|---|---|
| Qwen-Image | Qwen/Qwen-Image |
| FLUX.1-schnell | black-forest-labs/FLUX.1-schnell |
| SD3 Medium | stabilityai/stable-diffusion-3-medium-diffusers |
| Segmind-Vega | segmind/Segmind-Vega |
| SDXL 1.0 | stabilityai/stable-diffusion-xl-base-1.0 |
| Z-Image-Turbo | Tongyi-MAI/Z-Image-Turbo |
| 模型名称 | 模型标识 |
|---|---|
| Qwen-Image-Edit | Qwen/Qwen-Image-Edit |
| Qwen-Image-Edit-2509 | Qwen/Qwen-Image-Edit-2509 |
| SD Inpainting | runwayml/stable-diffusion-inpainting |
在 ModelScope 官网 注册账号，然后在个人中心获取 API 令牌。
- 检查 API 令牌是否正确
- 检查网络连接是否正常
- 检查提示词是否符合要求
- 尝试降低批次大小或增加超时时间
- 查看控制台输出的错误信息，根据错误信息进行排查
编辑 nodes/modelscope_api_model_presets_custom.json 文件，按照以下格式添加模型预设：
{
"model_name": "模型名称",
"model_id": "模型标识"
}
编辑 nodes/modelscope_api_lora_presets_custom.json 文件，按照以下格式添加 LoRA 模型预设：
{
"lora_name": "LoRA模型名称",
"lora_id": "LoRA模型标识"
}
支持的 LoRA 模型格式为 "model_name|weight"，其中 weight 是可选的，默认值为 1.0。例如：
- "qiyuanai/TikTok_Xiaohongshu_career_line_beauty_v1|0.8"
- "qiyuanai/TikTok_Xiaohongshu_career_line_beauty_v1"
本项目采用 MIT 许可证，详情请参考 LICENSE 文件。
欢迎提交 Issue 和 Pull Request 来改进本项目。
如有问题或建议，欢迎通过以下方式联系：
- 在 GitHub 上提交 Issue
- 发送邮件到项目维护者邮箱
- 初始版本
- 支持文生图功能
- 支持图像编辑功能
- 支持 API 密钥管理
- 支持模型预设管理
- 支持 LoRA 预设管理
