Skip to content

Commit 1535a2c

Browse files
authored
Disable quick init for TapasPreTrainedModel (#32149)
add attribute to model Signed-off-by: Daniel Lok <daniel.lok@databricks.com>
1 parent 34b4321 commit 1535a2c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/transformers/models/tapas/modeling_tapas.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -724,6 +724,7 @@ class TapasPreTrainedModel(PreTrainedModel):
724724
config_class = TapasConfig
725725
base_model_prefix = "tapas"
726726
supports_gradient_checkpointing = True
727+
_supports_param_buffer_assignment = False
727728

728729
# Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights
729730
def _init_weights(self, module):

0 commit comments

Comments
 (0)