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
22
23
24
25
26

use crate::DiffuserScheduler;
use serde::Serialize;

#[repr(u8)]
#[derive(Clone, Debug, Serialize)]
pub enum DiffuserSchedulerKind {
    Euler,
    DDIM,
}

mod der;

impl Default for DiffuserSchedulerKind {
    fn default() -> Self {
        Self::DDIM
    }
}

impl DiffuserSchedulerKind {
    pub fn as_scheduler(&self) -> DiffuserScheduler {
        match self {
            DiffuserSchedulerKind::Euler => DiffuserScheduler::Euler(Box::default()),
            DiffuserSchedulerKind::DDIM => DiffuserScheduler::DDIM(Box::default()),
        }
    }
}