Crate diffusion_rs_core

Core crate for interacting with diffusion_rs.

The API is intentionally straightforward but strives to provide strong flexibility.

use std::time::Instant;

use diffusion_rs_core::{DiffusionGenerationParams, ModelSource, ModelDType, Offloading, Pipeline, TokenSource};

let pipeline = Pipeline::load(
    ModelSource::dduf("FLUX.1-dev-Q4-bnb.dduf")?,
    true,
    TokenSource::CacheToken,
    None,
    None,
    &ModelDType::Auto,
)?;

let start = Instant::now();

let images = pipeline.forward(
    vec!["Draw a picture of a sunrise.".to_string()],
    DiffusionGenerationParams {
        height: 720,
        width: 1280,
        num_steps: 50,
        guidance_scale: 3.5,
    },
)?;

let end = Instant::now();
println!("Took: {:.2}s", end.duration_since(start).as_secs_f32());

images[0].save("image.png")?;

Structs

• DiffusionGenerationParams
  Generation parameters.
• Pipeline
  Represents the model and provides methods to load and interact with it.

Enums

• ModelDType
  DType for the model.
• ModelSource
  Source from which to load the model. This is easiest to create with the various constructor functions.
• Offloading
  Offloading setting during loading.
• TokenSource
  The source of the HF token.

Traits

• TryIntoDType
  Type which can be converted to a DType