stable_diffusion_trainer/prompt/
mod.rs

1//! Prompt module.
2
3use crate::prelude::*;
4
5/// The prompt structure.
6#[derive(Debug, Serialize, Deserialize)]
7pub struct Prompt {
8    /// The instance prompt.
9    pub instance: String,
10    /// The class prompt.
11    pub class: String
12}
13
14impl Prompt {
15    /// Create a new prompt structure.
16    pub fn new(instance: impl Into<String>, class: impl Into<String>) -> Self {
17        let instance = instance.into();
18        let class = class.into();
19        Prompt { instance, class }
20    }
21}