receive review comments and modify

OpenBMB · May 31, 2024 · a913ca4 · a913ca4
1 parent 88f5e6a
commit a913ca4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/examples/minicpmv/minicpmv.cpp b/examples/minicpmv/minicpmv.cpp
@@ -210,7 +210,7 @@ static bool bicubic_resize(const clip_image_u8 &img, clip_image_u8 &dst, int tar
             }
         }
     }
-    
+
     return true;
 }
 

diff --git a/examples/minicpmv/minicpmv_wrapper.cpp b/examples/minicpmv/minicpmv_wrapper.cpp
@@ -108,7 +108,7 @@ void process_image(struct minicpmv_context * ctx_llava, std::vector<std::vector<
 
     system_prompt = "<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\n";
     LOG_TEE("%s: image token past: %d\n", __func__, n_past);
-    eval_string(ctx_llava->ctx_llama, (system_prompt+"<image>").c_str(), params->n_batch, &n_past, true);
+    eval_string(ctx_llava->ctx_llama, (system_prompt+"<image>").c_str(), params->n_batch, &n_past, false);
     llava_eval_image_embed(ctx_llava->ctx_llama, image_embed_slices[0][0], params->n_batch, &n_past);
     eval_string(ctx_llava->ctx_llama, std::string("</image>").c_str(), params->n_batch, &n_past, false);
     if (image_embed_slices.size() > 1) {

diff --git a/examples/minicpmv/requirements.txt b/examples/minicpmv/requirements.txt
@@ -1,3 +1,4 @@
 -r ../../requirements/requirements-convert.txt
 pillow~=10.2.0
 torch~=2.1.1
+torchvision==0.16.2