riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[BigBird Pegasus] set _supports_param_buffer_assignment to False (#32222)

Browse Source 
set _supports_param_buffer_assignment to False
This commit is contained in:
Kashif Rasul 2024-07-25 15:11:43 +02:00 committed by GitHub
parent 1ecedf1d9e
commit 9b9a54e61b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py
View File

@ -1569,6 +1569,7 @@ class BigBirdPegasusPreTrainedModel(PreTrainedModel):
supports_gradient_checkpointing = True
_no_split_modules = ["BigBirdPegasusEncoderLayer", "BigBirdPegasusDecoderLayer"]
_skip_keys_device_placement = "past_key_values"
_supports_param_buffer_assignment = False
def _init_weights(self, module):
std = self.config.init_std