remove prints

StarlightSearch · Jan 30, 2025 · ab61898 · ab61898
1 parent d14d78a
commit ab61898
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/rust/src/embeddings/local/modernbert.rs b/rust/src/embeddings/local/modernbert.rs
@@ -107,10 +107,8 @@ impl BertEmbed for ModernBertEmbedder {
         let mut encodings: Vec<EmbeddingResult> = Vec::new();
 
         for mini_text_batch in text_batch.chunks(batch_size) {
-            println!("mini_text_batch: {:?}", mini_text_batch.len());
             let token_ids =
                 tokenize_batch(&self.tokenizer, mini_text_batch, &self.device)?;
-            println!("token_ids: {:?}", token_ids.shape());
             let attention_mask = get_attention_mask(&self.tokenizer, mini_text_batch, &self.device)?;
             let embeddings: Tensor = self.model.forward(&token_ids, &attention_mask)?;
             let pooled_output = self