abetlen / abetlen/llama-cpp-python

MiniCPM-V 2.6 memory leak occurred !!!

Ouverte
#1,886 5 commentaires 1 réaction 0 personnes assignées Voir sur GitHub
Langage dominant
Python
Étoiles
10.6k
Forks
1.4k
Métriques de merge des PR
Métriques de PR en attente

Description

# Prerequisites

Please answer the following questions for yourself before submitting an issue.

- [ ] I am running the latest code. Development is very rapid so there are no tagged versions as of now.
- [ ✅] I carefully followed the [README.md](https://github.com/abetlen/llama-cpp-python/blob/main/README.md).
- [ ✅] I [searched using keywords relevant to my issue](https://docs.github.com/en/issues/tracking-your-work-with-issues/filtering-and-searching-issues-and-pull-requests) to make sure that I am creating a new issue that is not already open (or closed).
- [ ✅] I reviewed the [Discussions](https://github.com/abetlen/llama-cpp-python/discussions), and have a new bug or useful enhancement to share.

# Expected Behavior
MiniCPM-V 2.6 memory leak ?

# Current Behavior
I am testing [minicpm-v-2.6](https://huggingface.co/openbmb/MiniCPM-V-2_6-gguf) ,here is part of my test code:
![37a5d00121d9619b47c64c17b35131b](https://github.com/user-attachments/assets/2163bd7a-962f-41e4-8ec5-773fa6a7026d)

when i run code ,it adds 10MB per inference,i use memory_profiler find in https://github.com/abetlen/llama-cpp-python/blob/2bc1d97c9672320828e70dc8293d5f8754682109/llama_cpp/llama_chat_format.py#L2839

![a6879c9732cf5cd576859e126b66faf](https://github.com/user-attachments/assets/070b8d5c-9471-4aa5-9f3d-4272c271ad06)

I thought at first that this variable "embed" wasn't being released,I added the code manual release on line 2856,but the program reported an error.
then I replace "embed" with "self._last_image_embed",Memory leaks continue to occur.
# Environment and Context

Please provide detailed information about your computer setup. This is important in case the issue is not reproducible except for under certain specific conditions.then

Example environment info:
```

Python 3.10.10
ubuntu 18.04
cudatoolkit 12.1
llama-cpp-python 0.2.90

```

Guide de contribution

Ouvrir le guide de contribution

Évaluation

Cette issue n'a pas encore été évaluée.

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.