Disable quick init for TapasPreTrainedModel (#32149)

add attribute to model Signed-off-by: Daniel Lok <daniel.lok@databricks.com>
huggingface · Jul 23, 2024 · 1535a2c · 1535a2c
1 parent 34b4321
commit 1535a2c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/models/tapas/modeling_tapas.py b/src/transformers/models/tapas/modeling_tapas.py
@@ -724,6 +724,7 @@ class TapasPreTrainedModel(PreTrainedModel):
     config_class = TapasConfig
     base_model_prefix = "tapas"
     supports_gradient_checkpointing = True
+    _supports_param_buffer_assignment = False
 
     # Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights
     def _init_weights(self, module):