[BugFix] Fix TypeError: unhashable type: 'dict' when serving deepseek32 (#30924)

Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
Lucas Wilkinson 2025-12-19 16:07:54 -05:00 committed by GitHub
parent 3bd8335bd0
commit 5f6477d1d0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1641,7 +1641,10 @@ class GPUModelRunner(
) -> None:
attn_group = self.attn_groups[kv_cache_gid][attn_gid]
builder = attn_group.get_metadata_builder(ubid or 0)
cache_key = (kv_cache_groups[kv_cache_gid].kv_cache_spec, type(builder))
kv_cache_spec = kv_cache_groups[kv_cache_gid].kv_cache_spec
if isinstance(kv_cache_spec, UniformTypeKVCacheSpecs):
kv_cache_spec = kv_cache_spec.kv_cache_specs[attn_group.layer_names[0]]
cache_key = (kv_cache_spec, type(builder))
cascade_attn_prefix_len = (
cascade_attn_prefix_lens[kv_cache_gid][attn_gid]