stable-diffusion-cli 0.1.9

Stable Diffusion CLI
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
34
35

#![doc=include_str!("../README.md")]

use clap::{Parser, Subcommand};

mod generate;
mod train;

#[derive(Debug, Parser, Clone)]
#[command(author, version, about, long_about = None)]
pub struct Arguments {
    #[command(subcommand)]
    command: Command
}

#[derive(Debug, Subcommand, Clone)]
#[command(author, version, about, long_about = None)]
pub enum Command {
    /// Generate a Stable Diffusion image.
    Generate(generate::Arguments),
    /// Train a Stable Diffusion model
    Train(train::Arguments),
}

impl Arguments {
    pub fn execute(self) -> anyhow::Result<()> {
        match self.command {
            Command::Generate(args) => args.execute(),
            Command::Train(args) => args.execute(),
        }
    }
}

fn main() -> anyhow::Result<()> {
    Arguments::parse().execute()
}