#ai #image #image-generation #models #candle #prompt #interface

rwuerstchen

A simple interface for RWuerstchen image generation models models

3 unstable releases

0.2.1 Feb 28, 2024
0.2.0 Feb 18, 2024
0.1.0 Dec 16, 2023

#612 in Images

22 downloads per month
Used in 2 crates (via kalosm-vision)

MIT/Apache and LGPL-3.0

26KB
444 lines

RWuerstchen

RWuerstchen is a rust wrapper for library for Wuerstchen implemented in the Candle ML framework.

RWuerstchen generates images efficiently from text prompts.

Usage

use rwuerstchen::*;

let model = Wuerstchen::builder().build().unwrap();
let settings = WuerstchenInferenceSettings::new(
    "a cute cat with a hat in a room covered with fur with incredible detail",
)
.with_n_steps(2);
let images = model.run(settings).unwrap();
for (i, img) in images.iter().enumerate() {
    img.save(&format!("{}.png", i)).unwrap();
}

Dependencies

~34–48MB
~698K SLoC