jonathanjordan21
commited on
Update modeling_qwen2_nomic_vision.py
Browse files
modeling_qwen2_nomic_vision.py
CHANGED
@@ -933,7 +933,7 @@ class Qwen2NomicVisionModel(Qwen2NomicVisionPreTrainedModel):
|
|
933 |
img_emb = self.vision_model(image).last_hidden_state
|
934 |
img_embeddings = F.normalize(img_emb[:, 0], p=2, dim=1)
|
935 |
mix = self.mix_linear(img_embeddings).unsqueeze(1)
|
936 |
-
hidden_states = torch.cat((
|
937 |
# hidden_states = hidden_states.concat()
|
938 |
|
939 |
# add hidden states from the last decoder layer
|
|
|
933 |
img_emb = self.vision_model(image).last_hidden_state
|
934 |
img_embeddings = F.normalize(img_emb[:, 0], p=2, dim=1)
|
935 |
mix = self.mix_linear(img_embeddings).unsqueeze(1)
|
936 |
+
hidden_states = torch.cat((hidden_states, mix), dim=1)
|
937 |
# hidden_states = hidden_states.concat()
|
938 |
|
939 |
# add hidden states from the last decoder layer
|