GoogleCloudPlatform / GoogleCloudPlatform/accelerated-platforms
Too large max_model_len for Gemma 27b on H100
Open
- Dominant language
- HCL
- Stars
- 103
- Forks
- 36
- Avg merge
- 4d 9h
- Merged PRs (30d)
- 5
Description
When trying to spin up the recipe for inference on H100 GPU, I got the following issue about GPU Memory usage indicating that the 128k max_model_len is too large for this model to fit into the GPU memory:
(EngineCore_DP0 pid=121) ValueError: To serve at least one request with the models's max seq len (131072), (13.66 GiB KV cache is needed, which is larger than the available KV cache memory (8.85 GiB). Based on the available memory, the estimated maximum model length is 68000. Try increasing `gpu_memory_utilization` or decreasing `max_model_len` when initializing the engine.\
Contributor guide
Assessment
This issue has not been assessed yet.