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

fix

Browse Source
This commit is contained in:
Arthur 2025-07-03 16:32:29 +02:00
parent c278e1cb1c
commit e3c82cb7ed
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/sam/modeling_sam.py
View File

@ -1029,7 +1029,7 @@ class SamVisionNeck(nn.Module):
class SamVisionEncoder(PreTrainedModel):
_can_record_outputs = {"hidden_states": (SamVisionLayer, 0), "attentions": (SamVisionAttention, 1)}
_can_record_outputs = {"hidden_states": SamVisionLayer, "attentions": SamVisionAttention}
def __init__(self, config: SamVisionConfig):
super().__init__(config)

6
src/transformers/models/starcoder2/modeling_starcoder2.py
View File

@ -300,9 +300,9 @@ class Starcoder2PreTrainedModel(PreTrainedModel):
_supports_quantized_cache = True
_supports_static_cache = True
_supports_attention_backend = True
_can_record_outputs: dict[str, tuple[nn.Module, int]] = {
"hidden_states": (Starcoder2DecoderLayer, 0),
"attentions": (Starcoder2Attention, 1),
_can_record_outputs: = {
"hidden_states": Starcoder2DecoderLayer,
"attentions": Starcoder2Attention,
}
def _init_weights(self, module):

4
src/transformers/models/t5gemma/modular_t5gemma.py
View File

@ -550,8 +550,8 @@ def make_default_2d_attention_mask(
class T5GemmaEncoder(T5GemmaPreTrainedModel):
_can_record_outputs = {
"attentions": (T5GemmaSelfAttention, 1),
"hidden_states": (T5GemmaEncoderLayer, 1),
"attentions": T5GemmaSelfAttention,
"hidden_states": T5GemmaEncoderLayer,
}
def __init__(self, config):