mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-01-14 10:04:28 +08:00
[BugFix] Fix TypeError: unhashable type: 'dict' when serving deepseek32 (#30924)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
parent
3bd8335bd0
commit
5f6477d1d0
@ -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]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user