Remove unused code in GPT-Neo (#24826)

1

Remove unused code in GPT-Neo (#24826)
1
18d42bfd · namespace-Pt · GitHub · 9771ad33 · 18d42bfd
Unverified Commit 18d42bfd authored Jul 17, 2023 by namespace-Pt Committed by GitHub Jul 17, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

src/transformers/models/gpt_neo/modeling_gpt_neo.py src/transformers/models/gpt_neo/modeling_gpt_neo.py +1 -2

No files found.
--- a/src/transformers/models/gpt_neo/modeling_gpt_neo.py
+++ b/src/transformers/models/gpt_neo/modeling_gpt_neo.py
@@ -696,8 +696,7 @@ class GPTNeoForCausalLM(GPTNeoPreTrainedModel):
            position_ids.masked_fill_(attention_mask == 0, 1)
            if past_key_values:
                position_ids = position_ids[:, -1].unsqueeze(-1)
-        else:
-            position_ids = None
+
        return {
            "input_ids": input_ids,
            "past_key_values": past_key_values,