From a94f233834a7f76e72e0adffa040ce30cd7e23b8 Mon Sep 17 00:00:00 2001 From: Joe Runde Date: Wed, 4 Feb 2026 14:33:05 -0700 Subject: [PATCH] :poop: Signed-off-by: Joe Runde --- vllm_spyre/v1/worker/spyre_model_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_spyre/v1/worker/spyre_model_runner.py b/vllm_spyre/v1/worker/spyre_model_runner.py index fa8ada4f8..06541a577 100644 --- a/vllm_spyre/v1/worker/spyre_model_runner.py +++ b/vllm_spyre/v1/worker/spyre_model_runner.py @@ -999,7 +999,7 @@ def _make_kv_cache_manager(self) -> FullAttentionManager: def _get_blocks(self, request_id: str) -> list[KVCacheBlock]: return self.kv_cache_manager.req_to_blocks[request_id] - def get_total_spyre_blocks(self) -> int: + def get_total_spyre_blocks(self) -> int: """Returns the total number of KV cache blocks available for spyre. This currently returns the number of blocks required for a full-sized batch, which may be greater than the available memory.