RWuerstchen

RWuerstchen 是一个 Rust 封装库，用于在 Wuerstchen 中实现，该库在 Candle 机器学习框架中。

RWuerstchen 可以高效地从文本提示中生成图像。

用法

use futures_util::StreamExt;
use rwuerstchen::*;
#[tokio::main]
async fn main() -> Result<(), anyhow::Error> {
    let model = Wuerstchen::builder().build().await?;
    let settings = WuerstchenInferenceSettings::new(
        "a cute cat with a hat in a room covered with fur with incredible detail",
    );
    if let Ok(mut images) = model.run(settings) {
        while let Some(image) = images.next().await {
            if let Some(buf) = image.generated_image() {
                buf.save(&format!("{}.png",image.sample_num()))?;
            }
        }
    }
    Ok(())
}