andrewqian123 commited on
Commit
e4bc263
·
verified ·
1 Parent(s): 0964e94

Update modeling_minicpmv.py

Browse files
Files changed (1) hide show
  1. 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, 10)
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 = []