biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[KVConnector] Keep KVTransferParams as a dict (#18033)

Browse Source
This commit is contained in:
Nick Hill
2025-05-14 08:05:57 -07:00
committed by GitHub
parent d066e52013
commit 59dd311cf5
7 changed files with 64 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vllm/v1/engine/core.py
View File

@@ -182,14 +182,10 @@ class EngineCore:
# Start grammar compilation asynchronously
self.structured_output_manager.grammar_init(req)
if req.raw_kv_transfer_params is not None:
if (kv_connector := self.scheduler.get_kv_connector()):
# Parse raw KV transfer params via connector.
kv_connector.set_kv_transfer_params(req)
else:
logger.warning(
"Got KVTransferParams, but no KVConnector found. "
"Disabling KVTransfer for this request.")
if req.kv_transfer_params is not None and (
not self.scheduler.get_kv_connector()):
logger.warning("Got kv_transfer_params, but no KVConnector found. "
"Disabling KVTransfer for this request.")
self.scheduler.add_request(req)