remove breakpoint and prints

modeling_flexbert.py

@@ -1705,12 +1705,11 @@ class FlexBertForCausalLM(FlexBertPreTrainedModel):
         #     # reshape to batch size
         #     logits = logits.view(-1, self.vocab_size)
         #     # NOTE: error from here above
-        breakpoint()
 
         if self.pad_logits:
-            print(f"Padding logits: {logits.shape}")
+            # print(f"Padding logits: {logits.shape}")
             new_logits = self.pad_inputs(logits, indices, batch_size, seq_len-1)[0]
-            print(f"New logits: {new_logits.shape}")
+            # print(f"New logits: {new_logits.shape}")
             # print(new_logits.shape)
             # if new_logits.dim() == 2:
             #     new_logits = new_logits.unsqueeze(0)