diff --git a/src/logits_processor.rs b/src/logits_processor.rs index 101cb7a..695538e 100644 --- a/src/logits_processor.rs +++ b/src/logits_processor.rs @@ -249,7 +249,6 @@ impl LogitsProcessor { let mut argsort_indices = (0..probs.len()).collect::>(); // Sort by descending probability. - dbg!(&probs); argsort_indices.sort_by(|&i, &j| probs[j].partial_cmp(&probs[i]).unwrap()); // Clamp smaller probabilities to zero.