stprobe 0.2.3

A minimal CLI for inspecting safetensors headers
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

use std::process::ExitCode;

use clap::Parser;

#[derive(Parser, Debug)]
#[command(
    name = "stprobe",
    version,
    about = "Inspect basic metadata from a .safetensors file",
    long_about = None
)]
struct Cli {
    /// Path or HTTP(S) URL to a .safetensors file
    file: String,
}

fn main() -> ExitCode {
    let cli = Cli::parse();

    match stprobe::inspect_input(&cli.file) {
        Ok(report) => {
            print!("{}", stprobe::render_report(&report));
            ExitCode::SUCCESS
        }
        Err(error) => {
            eprintln!("Error: {error}");
            ExitCode::FAILURE
        }
    }
}