From 4c00ac06fb1eda226819e8660eae7c1cdacfa9b9 Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Fri, 6 Oct 2023 00:54:56 +0800 Subject: [PATCH] fix(download): mark ggml model downloading should be optional, as ggml is only used for metal backend for now (#512) --- crates/tabby/src/download.rs | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/crates/tabby/src/download.rs b/crates/tabby/src/download.rs index 20e8721..9a8a94a 100644 --- a/crates/tabby/src/download.rs +++ b/crates/tabby/src/download.rs @@ -1,6 +1,6 @@ use clap::Args; use tabby_download::Downloader; -use tracing::info; +use tracing::{info, log::warn}; use crate::fatal; @@ -18,16 +18,15 @@ pub struct DownloadArgs { pub async fn main(args: &DownloadArgs) { let downloader = Downloader::new(&args.model, args.prefer_local_file); - let handler = |err| fatal!("Failed to fetch model '{}' due to '{}'", args.model, err,); - downloader .download_ctranslate2_files() .await - .unwrap_or_else(handler); + .unwrap_or_else(|err| fatal!("Failed to fetch model '{}' due to '{}'", args.model, err)); + downloader .download_ggml_files() .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); }