diff --git a/src/transformers/models/bart/modeling_bart.py b/src/transformers/models/bart/modeling_bart.py index 74119889c527..d14d5da0191c 100755 --- a/src/transformers/models/bart/modeling_bart.py +++ b/src/transformers/models/bart/modeling_bart.py @@ -645,7 +645,7 @@ def forward( ) # embed positions - positions = self.embed_positions(input, past_key_values_length, position_ids=position_ids) + positions = self.embed_positions(input_ids, past_key_values_length, position_ids=position_ids) positions = positions.to(inputs_embeds.device) hidden_states = inputs_embeds + positions diff --git a/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py b/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py index 30538b86ebf3..24b28123b22c 100755 --- a/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py +++ b/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py @@ -1820,7 +1820,7 @@ def forward( ) # embed positions - positions = self.embed_positions(input, past_key_values_length, position_ids=position_ids) + positions = self.embed_positions(input_ids, past_key_values_length, position_ids=position_ids) positions = positions.to(inputs_embeds.device) hidden_states = inputs_embeds + positions diff --git a/src/transformers/models/mbart/modeling_mbart.py b/src/transformers/models/mbart/modeling_mbart.py index 4766dc0b3e34..9284ddd5f21b 100755 --- a/src/transformers/models/mbart/modeling_mbart.py +++ b/src/transformers/models/mbart/modeling_mbart.py @@ -727,7 +727,7 @@ def forward( ) # embed positions - position_ids = self.embed_positions(input, past_key_values_length, position_ids=position_ids) + position_ids = self.embed_positions(input_ids, past_key_values_length, position_ids=position_ids) hidden_states = inputs_embeds + position_ids.to(inputs_embeds.device) hidden_states = self.layernorm_embedding(hidden_states) diff --git a/src/transformers/models/plbart/modeling_plbart.py b/src/transformers/models/plbart/modeling_plbart.py index 418e44d7029a..c189e32892ad 100644 --- a/src/transformers/models/plbart/modeling_plbart.py +++ b/src/transformers/models/plbart/modeling_plbart.py @@ -577,7 +577,7 @@ def forward( ) # embed positions - positions = self.embed_positions(input, past_key_values_length, position_ids=position_ids) + positions = self.embed_positions(input_ids, past_key_values_length, position_ids=position_ids) positions = positions.to(inputs_embeds.device) hidden_states = inputs_embeds + positions diff --git a/src/transformers/models/pp_formulanet/modeling_pp_formulanet.py b/src/transformers/models/pp_formulanet/modeling_pp_formulanet.py index 652d18a9a941..c08164d8ef91 100644 --- a/src/transformers/models/pp_formulanet/modeling_pp_formulanet.py +++ b/src/transformers/models/pp_formulanet/modeling_pp_formulanet.py @@ -909,7 +909,7 @@ def forward( ) # embed positions - position_ids = self.embed_positions(input, past_key_values_length, position_ids=position_ids) + position_ids = self.embed_positions(input_ids, past_key_values_length, position_ids=position_ids) hidden_states = inputs_embeds + position_ids.to(inputs_embeds.device) hidden_states = self.layernorm_embedding(hidden_states)