mirror of
https://github.com/ollama/ollama.git
synced 2025-05-10 18:06:33 +02:00
dbb149e6f7
Currently, the KV cache and graph are lazily allocated as needed. The cache is fully allocated on first use of the corresponding layer whereas the graph grows with the size of the context. This can be an issue if another application allocates more VRAM after we do our calculations - Ollama will crash in the middle of inference. If we instead allocate the maximum needed memory at startup of the runner, we will either succeed or fail at that point rather than at some surprising time in the future. Currently, this only generates a worst case batch for text, which means that vision models may get a partial allocation and continue to lazily allocate the rest. |
||
|---|---|---|
| .. | ||
| imageproc | ||
| input | ||
| models | ||
| testdata | ||
| model.go | ||
| model_test.go | ||
| process_text.go | ||
| process_text_spm.go | ||
| process_text_spm_test.go | ||
| process_text_test.go | ||