fix bug in sp

OpenBMB · Sep 27, 2024 · 3b46e1d · 3b46e1d
1 parent 549a22f
commit 3b46e1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/llava/minicpmv-cli.cpp b/examples/llava/minicpmv-cli.cpp
@@ -183,11 +183,11 @@ static void process_eval_image_embed_l(struct llava_context * ctx_llava, const s
 
     if (sp == 0) {
         std::memcpy(image_embed, buffer, token_len);
-        std::memcpy(image_embed + n_patches + 1, buffer+token_len, token_len);
+        std::memcpy(image_embed + (n_patches + 1) * token_len, buffer+token_len, token_len);
     }
     else if (sp == 1) {
         std::memcpy(image_embed, buffer+token_len*2, token_len);
-        std::memcpy(image_embed+ n_patches + 1, buffer+token_len*3, token_len);
+        std::memcpy(image_embed + (n_patches + 1) * token_len, buffer+token_len*3, token_len);
     }
 
     std::memcpy(image_embed+token_len, embeds->embed + idx * clip_n_patches(ctx_llava->ctx_clip) * clip_n_mmproj_embd(ctx_llava->ctx_clip), clip_embd_nbytes(ctx_llava->ctx_clip));