diffuser_scheduler/types/euler/
mod.rs

1use super::*;
2
3#[doc = include_str!("Readme.md")]
4#[derive(Clone, Debug, Serialize, Deserialize)]
5pub struct EulerDiscreteScheduler {
6    pub alphas_cumprod: f32,
7    pub sigmas: f32,
8    pub init_noise_sigma: f32,
9}
10
11impl Default for EulerDiscreteScheduler {
12    fn default() -> Self {
13        Self { alphas_cumprod: 0.0, sigmas: 0.0, init_noise_sigma: 0.0 }
14    }
15}
16
17impl From<EulerDiscreteScheduler> for DiffuserScheduler {
18    fn from(value: EulerDiscreteScheduler) -> Self {
19        DiffuserScheduler::Euler(Box::new(value))
20    }
21}