Merge pull request #59 from edgenai/feat/allocation-prediction

Feat/allocation prediction
edgenai · Mar 12, 2024 · 9c098a7 · 9c098a7
2 parents 9a21a8e + 43a6e69
commit 9c098a7
Show file tree

Hide file tree

Showing 7 changed files with 381 additions and 21 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/llama_cpp/src/detail.rs b/crates/llama_cpp/src/detail.rs
@@ -96,9 +96,9 @@ pub(crate) unsafe extern "C" fn llama_log_callback(
     };
 
     match level {
-        ggml_log_level_GGML_LOG_LEVEL_ERROR => error!("ggml: {text}"),
-        ggml_log_level_GGML_LOG_LEVEL_INFO => info!("ggml: {text}"),
-        ggml_log_level_GGML_LOG_LEVEL_WARN => warn!("ggml: {text}"),
+        ggml_log_level_GGML_LOG_LEVEL_ERROR => error!(target: "llama.cpp", "{text}"),
+        ggml_log_level_GGML_LOG_LEVEL_INFO => info!(target: "llama.cpp", "{text}"),
+        ggml_log_level_GGML_LOG_LEVEL_WARN => warn!(target: "llama.cpp", "{text}"),
         _ => trace!("ggml: {text}"),
     }
 }