modelscope-ng 0.2.0

A command-line interface tool for downloading models from Modelscope, which can also be used as a library
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use clap::Parser;
use modelscope_ng::ModelScope;
use std::env;
use std::path::PathBuf;

#[derive(Debug, Parser)]
#[command(version, about, long_about = None)]
struct Args {
    #[clap(subcommand)]
    command: SubCommand,
}

impl Args {
    fn default_save_dir() -> PathBuf {
        let path = env::home_dir().expect("Failed to get home directory");
        path.join(".modelscope").join("models")
    }
}

#[derive(Debug, Clone, Parser)]
enum SubCommand {
    /// Download model
    Download {
        /// Model ID
        #[arg(short, long)]
        model_id: String,
        /// The path to save the model, will be created if not exists
        #[arg(short, long, default_value_os_t = Args::default_save_dir())]
        save_dir: PathBuf,
    },
    /// Download a single file from a model
    DownloadFile {
        /// Model ID
        #[arg(short, long)]
        model_id: String,
        /// File path in the model repository
        #[arg(short, long)]
        file_path: String,
        /// The path to save the file, will be created if not exists
        #[arg(short, long, default_value_os_t = Args::default_save_dir())]
        save_dir: PathBuf,
    },
    /// Login to modelscope use your token
    Login {
        /// modelscope token
        #[arg(short, long)]
        token: String,
    },
    /// Logout
    Logout,
    /// List all local models
    List,
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let args = Args::parse();
    match args.command {
        SubCommand::Download { model_id, save_dir } => {
            ModelScope::download(&model_id, &save_dir).await?;
        }
        SubCommand::DownloadFile {
            model_id,
            file_path,
            save_dir,
        } => {
            ModelScope::download_single_file(&model_id, &file_path, &save_dir).await?;
        }
        SubCommand::Login { token } => {
            ModelScope::login(&token).await?;
        }
        SubCommand::Logout => {
            ModelScope::logout().await?;
        }
        SubCommand::List => {
            let models = ModelScope::list().await?;
            if models.is_empty() {
                println!();
                println!("No local models found.");
                println!();
            } else {
                println!();
                println!("Found {} local Models", models.len());
                println!();
                for (index, model) in models.iter().enumerate() {
                    println!("{:2}. {:<50} {}", index + 1, model.0, model.1);
                }
                println!();
            }
        }
    };

    Ok(())
}