[Model Runner V2] Fix kv_connector pre_forward order (#42676)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye
2026-05-15 08:13:59 -07:00
committed by GitHub
parent d792d993c1
commit af9616d845
+1 -1
View File
@@ -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():