From f46580964422c382ed5bc8229574cd92ae408c9c Mon Sep 17 00:00:00 2001 From: wenxinwang Date: Fri, 5 Dec 2025 19:05:00 +0800 Subject: [PATCH] [bugfix] Adapt the patch to support YAML sections. --- ucm/integration/vllm/patch/patch_funcs/v092/vllm_patch.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ucm/integration/vllm/patch/patch_funcs/v092/vllm_patch.py b/ucm/integration/vllm/patch/patch_funcs/v092/vllm_patch.py index 2a697efb0..0a9f03b40 100644 --- a/ucm/integration/vllm/patch/patch_funcs/v092/vllm_patch.py +++ b/ucm/integration/vllm/patch/patch_funcs/v092/vllm_patch.py @@ -553,14 +553,14 @@ def _patch_scheduler() -> None: from ucm.sparse.base import INVALID_SLOT, UcmSparseRole from ucm.sparse.state import ensure_ucm_sparse_initialized, get_ucm_sparse + from ucm.utils import Config def init_ucm_sparse(self): self.ucm_sparse = None if self.vllm_config.kv_transfer_config is not None: - if ( - "ucm_sparse_config" - in self.vllm_config.kv_transfer_config.kv_connector_extra_config - ): + ucm_config = Config(self.vllm_config.kv_transfer_config) + ucm_sparse_config = ucm_config.get_config().get("ucm_sparse_config") + if ucm_sparse_config: ensure_ucm_sparse_initialized( self.vllm_config, role=UcmSparseRole.SCHEDULER )