Skip to content

[Bugfix] Adapt UCM connector to vLLM kv_cache_config#955

Open
dante159753 wants to merge 2 commits into
ModelEngine-Group:developfrom
dante159753:ucm-kv-cache-config-adapt
Open

[Bugfix] Adapt UCM connector to vLLM kv_cache_config#955
dante159753 wants to merge 2 commits into
ModelEngine-Group:developfrom
dante159753:ucm-kv-cache-config-adapt

Conversation

@dante159753
Copy link
Copy Markdown
Contributor

@dante159753 dante159753 commented May 11, 2026

Purpose

now latest VLLM requires kv_cache_config in constructor, adapt UCM connector to it, without breaking the execution of older VLLM version.

Modifications

add method _init_kv_connector_base, check if it receives kv_cache_config param, then call corresponding init method

Test

@dante159753 dante159753 force-pushed the ucm-kv-cache-config-adapt branch 2 times, most recently from e322d6d to 87986b7 Compare May 11, 2026 08:33
@dante159753 dante159753 force-pushed the ucm-kv-cache-config-adapt branch from 87986b7 to ef25409 Compare May 11, 2026 08:46
@dante159753 dante159753 marked this pull request as ready for review May 11, 2026 08:51
def __init__(self, vllm_config, role):
def __init__(
self,
vllm_config,
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Missing type annotations for vllm_config and role parameters. All other connector classes use vllm_config: "VllmConfig", role: KVConnectorRole for consistency.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants