Oops removing print

allenai · Oct 11, 2024 · 2dccc4b · 2dccc4b
1 parent aea3f7f
commit 2dccc4b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pdelfin/train/dataprep.py b/pdelfin/train/dataprep.py
@@ -59,8 +59,6 @@ def prepare_data_for_qwen2_training(example, processor, add_batch_dim=False):
         padding=True,
         return_tensors="np"
     )
-
-    print(labels["input_ids"].shape)
 
     # Append an <|im_end|>\n" to the labels, because this is what it would look like
     # if we passed the whole message stream in there