Disclaimer Stable Diffusion is a trademark owned by Stability AI. Original repos: Stable Diffusion 1.5, Stable Diffusion 2.1, Stable Diffusion XL and XL-Turbo
This is the Stable Diffusion core library.
- Stable Diffusion 1.5
- Stable Diffusion 2.1
- Stable Diffusion XL
- Stable Diffusion Turbo
- CPU (default)
- MKL
- Cuda
- Metal
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(())
}