Skip to content

Commit 11ac9dd

Browse files
authored
Support all interleaved layer types (vllm-project#28485)
Signed-off-by: Yong Hoon Shin <[email protected]>
1 parent 5c9ad13 commit 11ac9dd

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

vllm/transformers_utils/config.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -472,8 +472,7 @@ def is_interleaved(config: PretrainedConfig) -> bool:
472472
"""
473473
text_config = config.get_text_config()
474474
if layer_types := getattr(text_config, "layer_types", None):
475-
interleaved_types = {"full_attention", "sliding_attention"}
476-
return interleaved_types.issubset(layer_types)
475+
return len(set(layer_types)) > 1
477476
return False
478477

479478

0 commit comments

Comments
 (0)