Upload folder using huggingface_hub
Browse files- modeling_intern_vit.py +1 -1
modeling_intern_vit.py
CHANGED
@@ -289,7 +289,7 @@ class InternVisionEncoderLayer(nn.Module):
|
|
289 |
"""
|
290 |
hidden_states = hidden_states + self.drop_path1(self.attn(self.norm1(hidden_states).to(hidden_states.dtype)) * self.ls1)
|
291 |
|
292 |
-
hidden_states = hidden_states + self.drop_path2(self.mlp(self.norm2(hidden_states)) * self.ls2)
|
293 |
|
294 |
return hidden_states
|
295 |
|
|
|
289 |
"""
|
290 |
hidden_states = hidden_states + self.drop_path1(self.attn(self.norm1(hidden_states).to(hidden_states.dtype)) * self.ls1)
|
291 |
|
292 |
+
hidden_states = hidden_states + self.drop_path2(self.mlp(self.norm2(hidden_states).to(hidden_states.dtype)) * self.ls2)
|
293 |
|
294 |
return hidden_states
|
295 |
|