stable_diffusion_trainer/model_file_format/
mod.rs

1//! Module for model file format enum.
2
3use std::fmt::Display;
4
5/// The model file format enumeration.
6pub enum ModelFileFormat {
7    /// The safetensors format.
8    Safetensors
9}
10
11impl Display for ModelFileFormat {
12    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
13        match self {
14            ModelFileFormat::Safetensors => write!(f, "safetensors")
15        }
16    }
17}