Disable quick init for TapasPreTrainedModel (#32149)

add attribute to model Signed-off-by: Daniel Lok <daniel.lok@databricks.com>

Disable quick init for TapasPreTrainedModel (#32149)
add attribute to model Signed-off-by: Daniel Lok <daniel.lok@databricks.com>
1535a2c9 · Daniel Lok · GitHub · 34b43211 · 1535a2c9
Unverified Commit 1535a2c9 authored Jul 23, 2024 by Daniel Lok Committed by GitHub Jul 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/models/tapas/modeling_tapas.py src/transformers/models/tapas/modeling_tapas.py +1 -0

No files found.
--- 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):