Merge pull request #39 from beclab/feat/bertv3-embedding

fix: make log level debug
beclab · Oct 21, 2024 · a50f233 · a50f233
2 parents 5b6b901 + 14aa208
commit a50f233
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/user-embedding/src/bertcommon.rs b/user-embedding/src/bertcommon.rs
@@ -137,7 +137,7 @@ async fn calculate_userembedding() -> AnyhowResult<Tensor, AnyhowError> {
             calculate_single_entry_pure(current_entry_id, &model, current_tokenizer).await;
         if let Some(current_tensor) = current_tensor_option {
             cumulative_tensor = cumulative_tensor.add(&current_tensor)?;
-            tracing::debug!("add current_entry {}", current_entry_id);
+            tracing::info!("add current_entry {}", current_entry_id);
         } else {
             tracing::error!("current_entry_id {} calculate fail", current_entry_id);
         }

diff --git a/user-embedding/src/common.rs b/user-embedding/src/common.rs
@@ -13,6 +13,7 @@ pub fn init_tracing() {
     tracing_subscriber::fmt()
         .with_line_number(true)
         .with_file(true)
+        .with_max_level(tracing::Level::DEBUG)
         .init();
     tracing::error!("tracing initialized.");
 }