Struct NextDit 

pub struct NextDit {
    pub hidden: usize,
    pub in_channels: usize,
    pub patch: usize,
    /* private fields */
}

Next-DiT: exact f32 from a diffusers directory, or CMF-quantized (mmap-resident) from a packaged file.

Fields

hidden: usize

in_channels: usize

patch: usize

Implementations

impl NextDit

pub fn load_dir(dir: &Path) -> Result<Self, String>

pub fn from_cmf(model: &Arc<CmfModel>) -> Result<Self, String>

Load from a packaged imagegen .cmf (dit.* tensors + dit.config_json). Quantized projections stay mmap-resident.

pub fn forward( &self, latent: &[f32], h: usize, w: usize, cap: &[f32], cap_n: usize, t: f32, ) -> Vec<f32>

One denoising forward: latent [c, h, w] (NCHW), caption features [cap_n, cap_feat], timestep t ∈ [0,1] (the pipeline’s 1 − σ). Returns the velocity prediction [c, h, w].

pub fn refine_caption(&self, cap: &[f32], cap_n: usize) -> Vec<f32>

Caption features → hidden, through the context refiner. Depends on NOTHING that moves during denoising — not the timestep, not the latents — so the whole thing is a constant of the prompt. The denoise loop hoists it out and hands the result to forward_with_cap; it used to be recomputed on every model call, which for 30 steps under CFG meant 60 evaluations of a value with two distinct instances.

pub fn forward_with_cap( &self, latent: &[f32], h: usize, w: usize, cap_e_in: &[f32], cap_n: usize, t: f32, ) -> Vec<f32>

The rest of the forward, from an already-refined caption.

Trait Implementations

impl Drop for NextDit

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.