# AISolution-ComfyUI

**Repository Path**: intel-china/aisolution-comfyui

## Basic Information

- **Project Name**: AISolution-ComfyUI
- **Description**: ComfyUI 是一个用户友好的图形界面，专为Stable Diffusion设计，是一个强大的深度学习模型，用于从文本提示生成图像。它支持多种插件，可以增强其功能。本示例展示了如何在 Intel Gaudi 硬件上部署 ComfyUI 应用程序。Intel Gaudi2D 加速器支持深度学习模型的训练和推理，更多细节请访问 https://habana.ai/products
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-03-04
- **Last Updated**: 2025-11-06

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README


# ComfyUI 应用程序

ComfyUI 是一个用户友好的图形界面，专为 **Stable Diffusion** 设计，这是一个强大的深度学习模型，用于从文本提示生成图像。ComfyUI 简化了与 **Stable Diffusion** 的交互过程，使用户能够轻松输入文本描述并自定义各种参数以生成高质量的图像。它支持多种插件，可以增强其功能，例如添加新的模型/特性、提升性能或与其他工具集成。

**Stable Diffusion** 本身是一个先进的文本到图像生成模型，它利用扩散过程基于输入文本创建详细且连贯的图像。由于其能够生成多样化且富有创意的输出，它在艺术家、设计师和任何希望从文本描述生成视觉内容的人群中广受欢迎。

ComfyUI 与 Stable Diffusion 一起，为用户提供了一个易于访问且高效的方式来探索 AI 生成图像的能力。

本示例展示了如何在 Intel Gaudi 硬件上部署 ComfyUI 应用程序。

## 准备 ComfyUI 服务

### 准备仓库
```bash
git clone https://gitee.com/intel-china/aisolution-comfyui
cd aisolution-comfyui
```

### 构建 ComfyUI 镜像

#### 选项 1：使用 Docker 构建镜像

```bash
export REGISTRY=opea/aisolution
export IMAGE_TAG=latest
docker build -t ${REGISTRY}/comfyui:${IMAGE_TAG} -f src/hpu_patches_f5847582/Dockerfile --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy .
```

#### 选项 2：使用 Docker Compose 构建镜像

```bash
export REGISTRY=opea/aisolution
export IMAGE_TAG=latest
export WORKSPACE=`pwd`
docker compose -f deployment/docker_image_build/docker_build_compose.yaml build --build-arg http_proxy=$http_proxy --build-arg https_proxy=$https_proxy
```

## 启动 ComfyUI 服务

### 选项 1：使用 Docker 设置 ComfyUI 服务

#### 在 Gaudi 上运行 ComfyUI 容器

```bash
ip_address=$(hostname -I | awk '{print $1}')
export no_proxy="${no_proxy},${ip_address},localhost,127.0.0.1"
## 如果您想使用特定卡片，替换 "HABANA_VISIBLE_DEVICES=all" 为 "HABANA_VISIBLE_DEVICES=<card_id>"

docker run -d -it --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e no_proxy=${no_proxy} -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --net=host --ipc=host --name comfyui ${REGISTRY}/comfyui:${IMAGE_TAG}
```

#### 使用 OPEA Text2image 微服务设置 ComfyUI

```bash
export HF_TOKEN={your_hf_token}
export MODEL=stable-diffusion-v1-5/stable-diffusion-v1-5
docker run --ipc=host -p 9379:9379 -e http_proxy=$http_proxy -e https_proxy=$https_proxy -e no_proxy=${no_proxy} -e HF_TOKEN=$HF_TOKEN -e MODEL=$MODEL opea/text2image:latest
```

>>**注意**：  
若要使用本地缓存的模型，请使用 `export MOUNT_DIR=<your_server_HF_HOME>` 传递缓存路径到 Docker 容器，使用 `-v ${MOUNT_DIR}:/data -e HF_HOME=/data`。

```bash
ip_address=$(hostname -I | awk '{print $1}')
export no_proxy="${no_proxy},${ip_address},localhost,127.0.0.1"
export MODEL=stable-diffusion-v1-5/stable-diffusion-v1-5
export MOUNT_DIR=$HOME/.cache/huggingface

docker run -d --ipc=host -p 9379:9379 -v ${MOUNT_DIR}:/data -e HF_HOME=/data -e http_proxy=$http_proxy  -e https_proxy=$https_proxy -e no_proxy=${no_proxy} -e MODEL=${MODEL} -e HF_ENDPOINT=https://hf-mirror.com  opea/text2image:latest
```

请参考 [模型下载](https://huggingface.co/docs/hub/en/models-downloading) 从 Huggingface 下载模型。

#### 检查结果

1. 在浏览器中打开 `http://{your_ip}:9375/`
2. 在 ComfyUI 中加载 [plugins/ComfyUI-Diffusers-Habana/examples/pipeline_diffusers_endpoint.json](src/plugins/ComfyUI-Diffusers-Habana/examples/pipeline_diffusers_endpoint.json)
3. 点击 `Queue Prompt` 生成图像

### 选项 2：使用 Docker Compose 设置 ComfyUI 服务

另外，您可以通过以下命令轻松运行这两个服务：

```bash
ip_address=$(hostname -I | awk '{print $1}')
export no_proxy="${no_proxy},${ip_address},localhost,127.0.0.1"
export HF_TOKEN=<your-hf-token>
export MODEL=stable-diffusion-v1-5/stable-diffusion-v1-5
export MOUNT_DIR=$HOME/.cache/huggingface/hub
## 如果您想使用特定卡片，请导出 card_id=<card_id>
## export card_id=<card_id>
cd deployment/docker_compose/gaudi
docker compose up -d
```

#### 检查结果

1. 在浏览器中打开 `http://{your_ip}:9375/`
2. 在 ComfyUI 中加载 [plugins/ComfyUI-Diffusers-Habana/examples/pipeline_diffusers_endpoint.json](src/plugins/ComfyUI-Diffusers-Habana/examples/pipeline_diffusers_endpoint.json)
3. 点击 `Queue Prompt` 生成图像