Properly check if all fused layers are in the list of targets (#12666)

Thanks @kylesayrs for catching this!
2025-12-25 21:08:53 +08:00 · 2025-02-03 06:42:18 +01:00 · 2025-02-03 06:42:18 +01:00 · c5932e5dac
commit c5932e5dac
parent 20579c0fae
1 changed files with 1 additions and 1 deletions
--- a/vllm/model_executor/layers/quantization/compressed_tensors/utils.py
+++ b/vllm/model_executor/layers/quantization/compressed_tensors/utils.py
@ -236,7 +236,7 @@ def _match_fused_layer(layer_name: str,
                               for type_suffix in possible_layer_types)

        if is_same_parent and is_matching_type and all(
-                '.'.join([parent_path, type_suffix])
+            (f"{parent_path}.{type_suffix}" in target_layers)
                for type_suffix in possible_layer_types):
            return target