From cbaed9a0c4511439e2e9f956f39957e7fdaa9cb1 Mon Sep 17 00:00:00 2001 From: Silas Marvin <19626586+SilasMarvin@users.noreply.github.com> Date: Sun, 2 Jun 2024 18:10:52 -0700 Subject: [PATCH] Clean up eprintln --- src/transformer_backends/llama_cpp/mod.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformer_backends/llama_cpp/mod.rs b/src/transformer_backends/llama_cpp/mod.rs index 0bc1820..3e5e758 100644 --- a/src/transformer_backends/llama_cpp/mod.rs +++ b/src/transformer_backends/llama_cpp/mod.rs @@ -13,7 +13,7 @@ use anyhow::Context; use hf_hub::api::sync::ApiBuilder; use serde::Deserialize; use serde_json::Value; -use tracing::instrument; +use tracing::{error, instrument}; mod model; use model::Model; @@ -47,6 +47,7 @@ impl LLaMACPP { .name .as_ref() .context("Please set `name` to use LLaMA.cpp")?; + error!("Loading in: {} - {}\nIf this model has not been loaded before it may take a few minutes to download it. Please hangtight.", configuration.model.repository, name); let repo = api.model(configuration.model.repository.to_owned()); let model_path = repo.get(name)?; let model = Model::new(model_path, &configuration)?;