Update vision_encoder.py
Browse files- vision_encoder.py +1 -1
vision_encoder.py
CHANGED
|
@@ -172,7 +172,7 @@ class VisionProjection(nn.Module):
|
|
| 172 |
def __init__(self):
|
| 173 |
super().__init__()
|
| 174 |
|
| 175 |
-
image_embedding_dim =
|
| 176 |
model_dim = 2048
|
| 177 |
hidden_dim = model_dim * 4
|
| 178 |
|
|
|
|
| 172 |
def __init__(self):
|
| 173 |
super().__init__()
|
| 174 |
|
| 175 |
+
image_embedding_dim = 2304
|
| 176 |
model_dim = 2048
|
| 177 |
hidden_dim = model_dim * 4
|
| 178 |
|