vllm-nightly kv-cache calculation seems to be off
#7
by retowyss - opened
Just checking if someone else can confirm this problem.
vllm version 0.20.1rc1.dev64+g5560cac7e
CUDA_VISIBLE_DEVICES=0,1 vllm serve mistralai/Mistral-Medium-3.5-128B --tensor-parallel-size 2 --tool-call-parser mistral --enable-auto-tool-choice --reasoning-parser mistral --max_num_batched_tokens 16384 --max_num_seqs 4 --gpu_memory_utilization 0.93 --served-model-name Mistral-Medium-3.5-128B --max-model-len auto --enable-sleep-mode --speculative_config '{
"model": "mistralai/Mistral-Medium-3.5-128B-EAGLE",
"num_speculative_tokens": 3,
"method": "eagle",
"max_model_len": "65536"
}'
And similar configs, I end up with at most like 100k kv-cache allocation (.97 utilization) but it's only using 84gb per card.
Anyway, thanks Mistral - model looks great!
retowyss changed discussion status to closed