[BigBird Pegasus] set _supports_param_buffer_assignment to False (#32222

) set _supports_param_buffer_assignment to False
huggingface · Jul 25, 2024 · 9b9a54e · 9b9a54e
1 parent 1ecedf1
commit 9b9a54e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py b/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py
@@ -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