diffuser-scheduler 0.0.0

Shared definition for diffuser schedulers
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

use super::*;

#[doc = include_str!("Readme.md")]
#[derive(Clone, Debug, Serialize)]
pub struct DDIMScheduler {
    pub alphas_cumprod: f32,
    pub final_alpha_cumprod: f32,
    pub init_noise_sigma: f32,
}

impl Default for DDIMScheduler {
    fn default() -> Self {
        Self { alphas_cumprod: 0.0, final_alpha_cumprod: 0.0, init_noise_sigma: 0.0 }
    }
}

mod der;