From 5b4cabb8e126e7d5ad72e9d3e8579d434af11354 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Thu, 25 Jul 2024 13:59:46 +0200 Subject: [PATCH] set _supports_param_buffer_assignment to False --- .../models/bigbird_pegasus/modeling_bigbird_pegasus.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py b/src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py index d1ba54213a0346..9f8e3cd19cd835 100755 --- 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