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
18
19
20
21

use super::*;

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

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

impl From<EulerDiscreteScheduler> for DiffuserScheduler {
    fn from(value: EulerDiscreteScheduler) -> Self {
        DiffuserScheduler::Euler(Box::new(value))
    }
}