Unverified Commit 11ac9ddd authored by Yong Hoon Shin's avatar Yong Hoon Shin Committed by GitHub
Browse files

Support all interleaved layer types (#28485)


Signed-off-by: default avatarYong Hoon Shin <yhshin@meta.com>
parent 5c9ad138
...@@ -472,8 +472,7 @@ def is_interleaved(config: PretrainedConfig) -> bool: ...@@ -472,8 +472,7 @@ def is_interleaved(config: PretrainedConfig) -> bool:
""" """
text_config = config.get_text_config() text_config = config.get_text_config()
if layer_types := getattr(text_config, "layer_types", None): if layer_types := getattr(text_config, "layer_types", None):
interleaved_types = {"full_attention", "sliding_attention"} return len(set(layer_types)) > 1
return interleaved_types.issubset(layer_types)
return False return False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment