remove debugging print

huggingface · Sep 10, 2024 · ab1e3c9 · ab1e3c9
1 parent 4acf9bc
commit ab1e3c9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/nanotron/helpers.py b/src/nanotron/helpers.py
@@ -53,7 +53,6 @@ def _vocab_size_with_padding(orig_vocab_size: int, pg_size: int, make_vocab_size
     multiple = make_vocab_size_divisible_by * pg_size
     after = int(ceil(orig_vocab_size / multiple) * multiple)
     if after != orig_vocab_size:
-        print("i'm in")
         log_rank(
             f"[Vocab Size Padding] Padded vocab (size: {orig_vocab_size}) with {after - orig_vocab_size} dummy tokens (new size: {after})",
             logger=logger,