riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Disable quick init for TapasPreTrainedModel (#32149)

Browse Source 
add attribute to model

Signed-off-by: Daniel Lok <daniel.lok@databricks.com>
This commit is contained in:
Daniel Lok 2024-07-23 17:26:00 +08:00 committed by GitHub
parent 34b43211d7
commit 1535a2c93d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/tapas/modeling_tapas.py
View File

@ -724,6 +724,7 @@ class TapasPreTrainedModel(PreTrainedModel):
config_class = TapasConfig config_class = TapasConfig
base_model_prefix = "tapas" base_model_prefix = "tapas"
supports_gradient_checkpointing = True supports_gradient_checkpointing = True
_supports_param_buffer_assignment = False
# Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights # Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights
def _init_weights(self, module): def _init_weights(self, module):