Quellcode durchsuchen

llaba : hotfix for llava-1.6 image number (#5495)

Co-authored-by: John <cmt-nct@users.noreply.github.com>
John vor 1 Jahr
Ursprung
Commit
7930a8a6e8
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      examples/llava/llava.cpp

+ 1 - 1
examples/llava/llava.cpp

@@ -100,7 +100,7 @@ static bool clip_llava_handle_patches(clip_ctx * ctx_clip, std::vector<float *>
     int num_patches_width  = grid_shape.first;  // grid 1-4
     int num_patches_width  = grid_shape.first;  // grid 1-4
     int num_patches_height = grid_shape.second; // grid 1-4
     int num_patches_height = grid_shape.second; // grid 1-4
 
 
-    const size_t num_images = num_patches_width + num_patches_height + 1;
+    const size_t num_images = num_patches_width * num_patches_height + 1;
 
 
     // TODO: size calculation is not calculated - it's only tens of MB
     // TODO: size calculation is not calculated - it's only tens of MB
     size_t ctx_size = 0;
     size_t ctx_size = 0;