[Bugfix] Don't int-sort attention-type names in --kv-cache-dtype-skip-layers for packed KV dtypes (#60364)
Signed-off-by: lobo235 <7@netlobo.com> Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
co-authored by
Harry Mellor
parent
e41d5d5543
commit
0fc0cefb4e
@@ -2285,9 +2285,11 @@ class EngineArgs:
|
||||
)
|
||||
|
||||
boundary = TurboQuantConfig.get_boundary_skip_layers(model_config)
|
||||
existing = set(cache_config.kv_cache_dtype_skip_layers)
|
||||
# Sorted so the order does not depend on set iteration: the list is
|
||||
# a CacheConfig hash factor. Entries can also be attention type
|
||||
# names (e.g. "sliding_window"), so sort them as strings.
|
||||
cache_config.kv_cache_dtype_skip_layers = sorted(
|
||||
existing | set(boundary), key=int
|
||||
set(cache_config.kv_cache_dtype_skip_layers) | set(boundary)
|
||||
)
|
||||
|
||||
ray_runtime_env = None
|
||||
|
||||
Reference in New Issue
Block a user