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

Co-authored-by: John <cmt-nct@users.noreply.github.com>
This commit is contained in:
John 2024-02-15 08:59:18 +01:00 committed by GitHub
parent 704359e299
commit 7930a8a6e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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_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
size_t ctx_size = 0;