4 releases

0.1.3 Nov 14, 2024
0.1.2 Nov 11, 2024
0.1.1 Nov 4, 2024
0.1.0 Nov 2, 2024

#145 in Graphics APIs

43 downloads per month

MIT license

4MB
83K SLoC

C 35K SLoC // 0.1% comments C++ 28K SLoC // 0.1% comments CUDA 7.5K SLoC // 0.0% comments Metal Shading Language 5K SLoC // 0.0% comments Objective-C 3K SLoC // 0.0% comments GLSL 2.5K SLoC // 0.0% comments Rust 2.5K SLoC // 0.0% comments Python 40 SLoC // 0.3% comments

diffusion-rs

Latest version Documentation

Rust bindings to https://github.com/leejet/stable-diffusion.cpp

Features Matrix

Windows Mac Linux
vulkan ✅️ ✅️ ✅️
metal ❌️ ✅️ ❌️
cuda ✅️ ❌️ ✅️
rocm ❔️ ❌️ ✅️
sycl ❔️ ❌️ ✅️

❔️: Not tested, should be supported

Usage

use diffusion_rs::{api::txt2img, preset::{Preset,PresetBuilder}};
let config = PresetBuilder::default()
            .preset(Preset::SDXLTurbo1_0Fp16)
            .prompt("a lovely duck drinking water from a bottle")
            .build()
            .unwrap();
txt2img(config).unwrap();

Troubleshooting

  • Something other than Windows/Linux isn't working!
    • I don't have a way to test these platforms, so I can't really help you.
  • I get a panic during binding generation build!
    • You can attempt to fix it yourself, or you can set the DIFFUSION_SKIP_BINDINGS environment variable. This skips attempting to build the bindings whatsoever and copies the existing ones. They may be out of date, but it's better than nothing.
      • DIFFUSION_SKIP_BINDINGS=1 cargo build
    • If you can fix the issue, please open a PR!

Roadmap

  1. Ensure that the underline cpp library compiles on supported platforms
  2. Build an easy to use library with model presets
  3. Automatic library publishing on crates.io by gh actions
  4. Maybe prebuilt CLI app binaries

Dependencies

~1–14MB
~121K SLoC