免责声明 Stable Diffusion 是 Stability AI 的商标。原始仓库：Stable Diffusion 1.5、Stable Diffusion 2.1、Stable Diffusion XL 和 XL-Turbo

Stable Diffusion

这是 Stable Diffusion 核心库。

支持版本

• Stable Diffusion 1.5
• Stable Diffusion 2.1
• Stable Diffusion XL
• Stable Diffusion Turbo

后端

• Cuda（默认 rust 功能）
• Metal
• Onnx
• CPU

示例

图像生成

use candle::Device;
use stable_diffusion::*;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let device = Device::new_cuda(0)?;
    let weights = StableDiffusionWeights::new(StableDiffusionVersion::XL, DType::F32);
    let parameters = StableDiffusionParameters::new(weights, device, DType::F16)?;
    let stable_diffusion = StableDiffusion::new(parameters)?;
    let args = GenerationParameters::new("A green apple");
    let image = stable_diffusion.generate(args)?;
    image.save("output.png")?;
    Ok(())
}