mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
fix
This commit is contained in:
parent
dc123e71ef
commit
37a1cb35b8
@ -642,12 +642,7 @@ class CLIPEncoder(nn.Module):
|
||||
def __init__(self, config: CLIPConfig, attn_implementation="eager"):
|
||||
super().__init__()
|
||||
self.config = config
|
||||
self.layers = nn.ModuleList(
|
||||
[
|
||||
CLIPEncoderLayer(config, attn_implementation=attn_implementation)
|
||||
for _ in range(config.num_hidden_layers)
|
||||
]
|
||||
)
|
||||
self.layers = nn.ModuleList([CLIPEncoderLayer(config) for _ in range(config.num_hidden_layers)])
|
||||
self.gradient_checkpointing = False
|
||||
|
||||
def forward(
|
||||
|
Loading…
Reference in New Issue
Block a user