cmfy 0.3.0

A CLI companion app for Comfy UI
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
27
28
29
30
31
32
33

use crate::{
    dto::{ClassType, PromptNodes},
    Result,
};
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct KSamplerNode {
    pub cfg: f32,
    pub denoise: f32,
    pub sampler_name: String,
    pub scheduler: String,
    pub steps: u8,
    pub seed: u64,
    #[serde(flatten)]
    pub other: serde_json::Value,
}

impl ClassType for KSamplerNode {
    const CLASS_TYPE: &str = "KSampler";
}

pub trait KSampler {
    fn reseed(&mut self) -> Result<()>;
}

impl KSampler for PromptNodes {
    fn reseed(&mut self) -> Result<()> {
        self.change_first_by_class(|sampler: &mut KSamplerNode| {
            sampler.seed = rand::random();
        })
    }
}