🖼️ Stable Diffusion 3

Stable Diffusion 3 模型可以在Hugging Face上面下载,用http协议clone下了提示没有权限,所以用的ssh方式克隆仓库

下载模型

由于,系统是Ubuntu,源里面的git-lfs太旧了,直接安装使用,后续克隆仓库会报错,要去git-lfs找软件源的地址,添加源地址后安装新版本的git-lfs
官网没有安装教程,教程在GitHub仓库文档

1
2
curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
sudo apt-get install git-lfs

clone model

1
2
git lfs install
git clone git@hf.co:stabilityai/stable-diffusion-3-medium

由于模型比较大,一次同步可能失败,失败后,直接在仓库目录下面执行git lfs pull继续同步

lfs clone

显卡这里有4个,选择性能好一点的运行模型
graphic card

为了能够直接显示图片,这里用ipynb运行,按照官网例子:

运行

1
2
3
4
5
6
7
8
9
10
11
12
13
import torch
from diffusers import StableDiffusion3Pipeline

pipe = StableDiffusion3Pipeline.from_pretrained("/root/ml/model/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
pipe = pipe.to("cuda:1")

image = pipe(
"A cat holding a sign that says hello world",
negative_prompt="",
num_inference_steps=28,
guidance_scale=7.0,
).images[0]
image

result

output