root
/
tabby
mirror of https://gitee.com/zhang_1334717033/tabby.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(download): mark ggml model downloading should be optional, as ggml is only used for metal backend for now (#512)

wsxiaoys-patch-1
Meng Zhang 2023-10-06 00:54:56 +08:00 committed by GitHub
parent 9cd2accbaa
commit 4c00ac06fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
crates/tabby/src/download.rs
View File

@ -1,6 +1,6 @@
use clap::Args; use clap::Args;
use tabby_download::Downloader; use tabby_download::Downloader;
use tracing::info; use tracing::{info, log::warn};
use crate::fatal; use crate::fatal;
@ -18,16 +18,15 @@ pub struct DownloadArgs {
pub async fn main(args: &DownloadArgs) { pub async fn main(args: &DownloadArgs) {
let downloader = Downloader::new(&args.model, args.prefer_local_file); let downloader = Downloader::new(&args.model, args.prefer_local_file);
let handler = |err| fatal!("Failed to fetch model '{}' due to '{}'", args.model, err,);
downloader downloader
.download_ctranslate2_files() .download_ctranslate2_files()
.await .await
.unwrap_or_else(handler); .unwrap_or_else(|err| fatal!("Failed to fetch model '{}' due to '{}'", args.model, err));
downloader downloader
.download_ggml_files() .download_ggml_files()
.await .await
.unwrap_or_else(handler); .unwrap_or_else(|err| warn!("Failed to fetch model '{}' due to '{}'", args.model, err));
info!("model '{}' is ready", args.model); info!("model '{}' is ready", args.model);
} }