Stable Diffusion 3.5 是 Stability AI 的 Text-to-Image 模型系列。Medium 约 2.5B 参数,强调质量与效率平衡;Large 约 8.1B 参数,强调更高质量与 Prompt 遵循能力。
安装
Shell
pip install -U torch diffusers transformers accelerate safetensors sentencepiece huggingface_hub
hf auth loginPython
import torch
from diffusers import StableDiffusion3Pipeline
pipe = StableDiffusion3Pipeline.from_pretrained(
"stabilityai/stable-diffusion-3.5-medium",
torch_dtype=torch.bfloat16,
).to("cuda")
image = pipe("A cinematic futuristic city", num_inference_steps=28, guidance_scale=5.0).images[0]
image.save("output.png")运行 Large 时将 Model ID 改为 stabilityai/stable-diffusion-3.5-large,显存不足时可使用 CPU offload、quantization 或降低分辨率。
ComfyUI
checkpoint 放入 ComfyUI/models/checkpoints/,clip_l、clip_g、t5xxl 放入 ComfyUI/models/text_encoders/。
Large 提供官方 Canny、Depth、Blur ControlNet。