Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +1 -1
modeling_minicpmv.py
CHANGED
|
@@ -293,7 +293,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 293 |
|
| 294 |
# Retrieve the embedding vector for pad_token_id
|
| 295 |
pad_embedding_vector = embedding_layer.weight[0]
|
| 296 |
-
vector_reshaped = pad_embedding_vector.view(1, 1,
|
| 297 |
|
| 298 |
for tensor in batch:
|
| 299 |
to_add = []
|
|
|
|
| 293 |
|
| 294 |
# Retrieve the embedding vector for pad_token_id
|
| 295 |
pad_embedding_vector = embedding_layer.weight[0]
|
| 296 |
+
vector_reshaped = pad_embedding_vector.view(1, 1, 4096)
|
| 297 |
|
| 298 |
for tensor in batch:
|
| 299 |
to_add = []
|