diff options
| author | Elbios <141279586+Elbios@users.noreply.github.com> | 2024-02-15 09:01:57 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-15 10:01:57 +0200 |
| commit | 0d4177126b0556e202efb85bf3f768be81076400 (patch) | |
| tree | b696a05bc6a6378b72cd0be3f96211ffe0056db1 /examples/llava/clip.h | |
| parent | 7930a8a6e89a04c77c51e3ae5dc1cd8e845b6b8f (diff) | |
llava : fix memory management bug (#5491)
* Fix memory management in llava and server code
Fixes this error:
llama_new_context_with_model: graph splits (measure): 3
Available slots:
-> Slot 0 - max context: 6000
{"timestamp":1707926446,"level":"INFO","function":"main","line":2623,"message":"model loaded"}
all slots are idle and system prompt is empty, clear the KV cache
slot 0 - loaded image
slot 0 is processing [task id: 0]
slot 0 : kv cache rm - [0, end)
slot 0 - encoding image [id: 1]
munmap_chunk(): invalid pointer
Aborted
* Make it cleaner by checking size in batch free wrapper
Diffstat (limited to 'examples/llava/clip.h')
| -rw-r--r-- | examples/llava/clip.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/examples/llava/clip.h b/examples/llava/clip.h index cd9a4022..e5bd5492 100644 --- a/examples/llava/clip.h +++ b/examples/llava/clip.h @@ -60,6 +60,8 @@ CLIP_API struct clip_image_f32 * clip_image_f32_init(); CLIP_API void clip_image_u8_free (struct clip_image_u8 * img); CLIP_API void clip_image_f32_free(struct clip_image_f32 * img); +CLIP_API void clip_image_u8_batch_free (struct clip_image_u8_batch & batch); +CLIP_API void clip_image_f32_batch_free(struct clip_image_f32_batch & batch); CLIP_API bool clip_image_load_from_file(const char * fname, struct clip_image_u8 * img); |