forked from Karylab-cklius/vllm
[Model Runner V2] Fix kv_connector pre_forward order (#42676)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
@@ -65,8 +65,8 @@ class ActiveKVConnector(KVConnector):
|
||||
|
||||
kv_connector_metadata = scheduler_output.kv_connector_metadata
|
||||
assert kv_connector_metadata is not None
|
||||
self.kv_connector.bind_connector_metadata(kv_connector_metadata)
|
||||
self.kv_connector.handle_preemptions(kv_connector_metadata)
|
||||
self.kv_connector.bind_connector_metadata(kv_connector_metadata)
|
||||
|
||||
# TODO: sort out KV Connectors' use of forward_context
|
||||
if is_forward_context_available():
|
||||
|
||||
Reference in New Issue
Block a user