当前位置：首页 > article >正文

部署stable-diffusion3.5 大模型，文生图

article 2024/11/8 6:49:49

UI 使用推荐的ComfyUI，GitHub 地址，huggingface 需要注册登录，需要下载的文件下面有说明

Dockerfile 文件如下：

FROM nvidia/cuda:12.4.0-base-ubuntu22.04
RUN apt-get update && apt-get install python3 pip git --no-install-recommends -y &&\
    git clone https://github.com/comfyanonymous/ComfyUI.git &&\
    pip install torch torchvision torchaudio --index-url https://pypi.tuna.tsinghua.edu.cn/simple --no-cache &&\
    apt-get clean && apt-get autoclean && apt-get autoremove
WORKDIR /ComfyUI
EXPOSE 8188
RUN pip install --index-url https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt --no-cache
COPY AIGODLIKE-COMFYUI-TRANSLATION custom_nodes/AIGODLIKE-COMFYUI-TRANSLATION
COPY stable-diffusion/sd3.5_large_turbo.safetensors models/checkpoints/
COPY stable-diffusion/vae models/vae
COPY clip models/clip
CMD ["python3", "main.py", "--listen", "0.0.0.0"]

把需要的文件下载下来拷贝到一个目录，文件夹目录结构如下：

.
├── AIGODLIKE-COMFYUI-TRANSLATION   # 中文包
│   ├── example
│   ├── __init__.py
│   ├── ja-JP
│   ├── ko-KR
│   ├── LocaleMap.js
│   ├── main.js
│   ├── MenuTranslate.js
│   ├── pyproject.toml
│   ├── readme.md
│   ├── ru-RU
│   ├── zh-CN
│   └── zh-TW
├── clip  # 解析器
│   ├── clip_g.safetensors
│   ├── clip_l.safetensors
│   └── t5xxl_fp8_e4m3fn.safetensors
├── Dockerfile
└── stable-diffusion ## 大模型
    ├── sd3.5_large_turbo.safetensors
    ├── SD3.5L_Turbo_example_workflow.json
    └── vae

需要下载大模型文件，demo 文件，vae 大模型地址
文本解析器：解析器地址
语言包：语言包地址，点击设置可以切换语言到中文，如果不需要可以忽略

构建镜像，并运行

docker build -t stable-diffusion:3.5-large-turbo-ubuntu22.04-cuda12.4-py310-torch2.5.1 .
docker run --runtime=nvidia --gpus all --rm -p 8188:8188 stable-diffusion:3.5-large-turbo-ubuntu22.04-cuda12.4-py310-torch2.5.1