Skip to content

GitLab

Unverified Commit fe2877ce authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

Remove unneeded prints in modeling_gpt_neox.py (#27080)

parent efba1a17
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
src/transformers/models/gpt_neox/modeling_gpt_neox.py
View file @ fe2877ce
...@@ -804,9 +804,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel): ...@@ -804,9 +804,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel):
self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs
): ):
input_shape = input_ids.shape 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 # cut decoder_input_ids if past is used
if past_key_values is not None: if past_key_values is not None:
past_length = past_key_values[0][0].shape[2] past_length = past_key_values[0][0].shape[2]
...@@ -837,7 +834,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel): ...@@ -837,7 +834,6 @@ class GPTNeoXForCausalLM(GPTNeoXPreTrainedModel):
model_inputs = {"inputs_embeds": inputs_embeds} model_inputs = {"inputs_embeds": inputs_embeds}
else: else:
model_inputs = {"input_ids": input_ids} model_inputs = {"input_ids": input_ids}
print(position_ids.shape)
model_inputs.update( model_inputs.update(
{ {
"attention_mask": attention_mask, "attention_mask": attention_mask,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment