Remove unneeded prints in modeling_gpt_neox.py (#27080)

src/transformers/models/gpt_neox/modeling_gpt_neox.py

@@ -804,9 +804,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel):
         self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs
     ):
         input_shape = input_ids.shape
-        print(input_shape)
-        print(past_key_values[0][0].shape if past_key_values is not None else "no pkv")
-
         # cut decoder_input_ids if past is used
         if past_key_values is not None:
             past_length = past_key_values[0][0].shape[2]
@@ -837,7 +834,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel):
             model_inputs = {"inputs_embeds": inputs_embeds}
         else:
             model_inputs = {"input_ids": input_ids}
-        print(position_ids.shape)
         model_inputs.update(
             {
                 "attention_mask": attention_mask,