/ src / model / params / img2img.rs
img2img.rs
 1  use image::DynamicImage;
 2  
 3  #[derive(Debug)]
 4  pub struct ImageToImageParams {
 5      pub prompt: String,
 6      pub negative_prompt: Option<String>,
 7      pub width: Option<u32>,
 8      pub height: Option<u32>,
 9      pub image: DynamicImage,
10      pub strength: f32,
11      pub steps: usize,
12      pub guidance_scale: f32,
13      pub seed: Option<u64>,
14  }
15  
16  impl Default for ImageToImageParams {
17      fn default() -> Self {
18          Self {
19              prompt: String::new(),
20              negative_prompt: None,
21              width: None, 
22              height: None,
23              image: DynamicImage::new_rgb8(1024, 1024),
24              strength: 0.75,
25              steps: 40,
26              guidance_scale: 7.5,
27              seed: None,
28          }
29      }
30  }