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

fix

Browse Source
This commit is contained in:
sayakpaul 2024-05-10 13:47:46 +02:00
parent dc123e71ef
commit 37a1cb35b8
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/models/clip/modeling_clip.py
View File

@ -642,12 +642,7 @@ class CLIPEncoder(nn.Module):
def __init__(self, config: CLIPConfig, attn_implementation="eager"): def __init__(self, config: CLIPConfig, attn_implementation="eager"):
super().__init__() super().__init__()
self.config = config self.config = config
self.layers = nn.ModuleList( self.layers = nn.ModuleList([CLIPEncoderLayer(config) for _ in range(config.num_hidden_layers)])
[
CLIPEncoderLayer(config, attn_implementation=attn_implementation)
for _ in range(config.num_hidden_layers)
]
)
self.gradient_checkpointing = False self.gradient_checkpointing = False
def forward( def forward(