Skip to content

Commit c3f7c1b

Browse files
committed
[Fix] Skip update connector output when no kv sending
1 parent 83a5650 commit c3f7c1b

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ucm/integration/vllm/uc_connector.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -767,7 +767,9 @@ def update_connector_output(self, connector_output: KVConnectorOutput):
767767
connector_output (KVConnectorOutput): the worker-side
768768
connectors output.
769769
"""
770-
done_sending = list(connector_output.finished_sending or set())
770+
if not connector_output.finished_sending:
771+
return
772+
done_sending = list(connector_output.finished_sending)
771773
self.connector.commit(done_sending, True)
772774
self.succeed_dumped_blocks.update(done_sending)
773775
connector_output.finished_sending = set()

0 commit comments

Comments
 (0)