stable-diffusion-webui这是目前最流行的 Stable Diffusion 图形界面(也支持命令行)。
先装:
# Ubuntu 示例
sudo apt update
sudo apt install -y python3.10 python3.10-venv gitgit clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui./webui.sh它会自动:
启动后访问:
http://127.0.0.1:7860diffusers适合写脚本、自动化。
pip install diffusers transformers accelerate torchfrom diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5"
).to("cuda")
image = pipe("a cat wearing sunglasses").images[0]
image.save("cat.png")模型通常放在:
models/Stable-diffusion/常用模型:
wget https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt❓ 没显卡能跑吗?
✅ 能,但很慢(CPU)
❓ 报错 CUDA?
✅ 说明 PyTorch 和驱动不匹配
❓ 模型太大?
pruned 版本如果你告诉我:
我可以直接给你一条一条可复制的命令。