企业 - 快速搭建Comfy UI

场景

小王是一家互联网公司的 AI 工程师，他希望利用公司的共享算力平台，快速搭建一个基于 Stable Diffusion 的图像生成平台，并通过 API 提供给公司系统封装使用。

1. 注册并创建实例

小王在NiceGPU注册账号后，使用5 NCU创建一个配置为4张NVIDIA RTX 4090的实例, 使用ComfyUI应用模板。

1. 什么是NCU？
   • 1 NCU 等于您可以在我们平台上连续使用24小时的一块A100 GPU的算力。
2. 针对NCU的使用场景和对应任务量举例：
   • 高性能计算任务：
     • 任务： 进行科学计算、仿真模拟等高性能计算任务。
     • 所需NCU： 5 NCU以上。高性能计算任务对计算速度和精度要求较高，需要强大的算力支持。
   • 大规模模型训练：
     • 任务： 训练大型语言模型、计算机视觉模型等。
     • 所需NCU： 10 NCU以上。大模型参数量巨大，训练时间长，需要大量的计算资源。
   • 分布式训练：
     • 任务： 将模型训练任务分布到多个GPU上，加速训练过程。
     • 所需NCU： 根据模型规模和分布式策略而定，通常需要多个NCU。
3. 影响NCU需求的因素:
   • 模型复杂度： 模型参数量越大，计算量越大，所需的NCU越多。
   • 数据集大小： 数据集越大，训练时间越长，所需的NCU越多。
   • 训练精度： 要求更高的训练精度，通常需要更多的训练迭代次数，从而消耗更多的计算资源。
   • 模型优化算法： 不同的优化算法对计算资源的需求也不同。

2. 创建自定义工作流

小王在 ComfyUI 中创建了一个自定义的工作流，包括： - 文本提示工程： 设计了一套详细的提示词模板，以生成不同风格的图像。 - 模型选择： 根据不同的需求，选择合适的 Stable Diffusion 模型。 - 图像后处理： 添加图像去噪、超分辨率等后处理步骤，提高图像质量。

3. 部署API

小王在 ComfyUI 的设置中开启了 API 功能，并设置了访问的端口和密钥。

curl http://localhost:7860/api/generate/images \
  -H "Authorization: Bearer your_api_key" \
  -X POST \
  -d '{"prompt": "a cat playing chess"}'

4. 提供AI 图像生成服务

5. 总结

通过共享算力平台和 ComfyUI，小王快速搭建了一个功能强大的图像生成平台，为公司提供了便捷的 AI 图像生成服务。