remove print

modeling_flexbert.py

@@ -1657,7 +1657,7 @@ class FlexBertForCausalLM(FlexBertPreTrainedModel):
             cu_seqlens=cu_seqlens,
             max_seqlen=max_seqlen,
         )
-        print(hidden_states.shape)
+        # print(hidden_states.shape)
 
         if self.compile_model:
             logits = self.compiled_lm_head(hidden_states)
@@ -1717,7 +1717,7 @@ class FlexBertForCausalLM(FlexBertPreTrainedModel):
                 attentions=None,
             )
         else:
-            print(f"Non-padding logits: {logits.shape}")
+            # print(f"Non-padding logits: {logits.shape}")
             # if logits.dim() == 2:
             #     logits = logits.unsqueeze(0)
             return CausalLMOutput(