Update vision_encoder.py
Browse files- vision_encoder.py +1 -1
vision_encoder.py
CHANGED
|
@@ -127,7 +127,7 @@ class LinearPatchEmbedding(nn.Module):
|
|
| 127 |
|
| 128 |
def __init__(self):
|
| 129 |
super().__init__()
|
| 130 |
-
self.linear = nn.Linear(588,
|
| 131 |
|
| 132 |
def forward(self, x):
|
| 133 |
b, c, hp1, wp2 = x.shape
|
|
|
|
| 127 |
|
| 128 |
def __init__(self):
|
| 129 |
super().__init__()
|
| 130 |
+
self.linear = nn.Linear(588, 4608)
|
| 131 |
|
| 132 |
def forward(self, x):
|
| 133 |
b, c, hp1, wp2 = x.shape
|