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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use super::Run;
use clap::Args;
use cmfy::{Client, Result};
use colored::Colorize;
use humansize::{make_format, BINARY};

/// Displays basic statistics about client and server.
#[derive(Debug, Args)]
pub struct Stats;

impl Run for Stats {
    async fn run(self, client: Client) -> Result<()> {
        println!("{}", "client".yellow());
        println!("    name        : {}", env!("CARGO_PKG_NAME"));
        println!("    version     : {}", env!("CARGO_PKG_VERSION"));
        println!("    client_id   : {}", client.id);

        let stats = client.system_stats().await?;
        println!("{}", "server".yellow());
        println!("    url         : {}", client.base_url()?);
        println!("    os          : {}", stats.system.os);
        println!("    versions");
        println!(
            "        python  : {}{}",
            stats
                .system
                .python_version
                .split_whitespace()
                .next()
                .expect("malfored python version"),
            if stats.system.embedded_python { " (embedded)" } else { "" }
        );
        println!("        comfyui : {}", stats.system.comfyui_version);
        println!("        pytorch : {}", stats.system.pytorch_version);
        println!("    devices");
        let format_size = make_format(BINARY);
        for (index, device) in stats.devices.iter().enumerate() {
            println!(
                "        {:<8}: {} ({}/{})",
                format!("[{}]", index),
                device.name,
                format_size(device.vram_free),
                format_size(device.vram_total),
            );
        }
        Ok(())
    }
}