From e5cf12e05268c1fbd3dfb277367239b6f2c4ad38 Mon Sep 17 00:00:00 2001 From: Fynn Schmitt-Ulms Date: Wed, 10 Sep 2025 15:44:04 -0400 Subject: [PATCH] Replace uses of deprecated `find_name_or_class_matches` fn Signed-off-by: Fynn Schmitt-Ulms --- src/llmcompressor/pipelines/layer_sequential/helpers.py | 4 ++-- src/llmcompressor/pipelines/sequential/helpers.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/llmcompressor/pipelines/layer_sequential/helpers.py b/src/llmcompressor/pipelines/layer_sequential/helpers.py index c8b9fcbd3..6b12760b4 100644 --- a/src/llmcompressor/pipelines/layer_sequential/helpers.py +++ b/src/llmcompressor/pipelines/layer_sequential/helpers.py @@ -5,7 +5,7 @@ import torch import tqdm -from compressed_tensors.quantization import find_name_or_class_matches +from compressed_tensors.utils.match import match_targets from torch.nn import Module from torch.utils.data.dataloader import DataLoader @@ -33,7 +33,7 @@ def match_modules(model: Module, target_names: List[str]) -> List[Module]: names_layers = [ (name, module) for name, module in model.named_modules() - if find_name_or_class_matches(name, module, target_names) + if match_targets(name, module, target_names) ] names_layers = sorted(names_layers, key=lambda name_layer: name_layer[0]) diff --git a/src/llmcompressor/pipelines/sequential/helpers.py b/src/llmcompressor/pipelines/sequential/helpers.py index f50795543..3f126efd0 100644 --- a/src/llmcompressor/pipelines/sequential/helpers.py +++ b/src/llmcompressor/pipelines/sequential/helpers.py @@ -6,12 +6,12 @@ import torch from accelerate.hooks import remove_hook_from_module -from compressed_tensors.quantization import find_name_or_class_matches from compressed_tensors.utils import ( has_offloaded_params, offloaded_dispatch, remove_dispatch, ) +from compressed_tensors.utils.match import match_targets from loguru import logger from torch.fx import Graph, GraphModule, Node from torch.fx.graph import PythonCode @@ -424,7 +424,7 @@ def match_modules(model: Module, target_names: List[str]) -> Set[Module]: return set( module for name, module in model.named_modules() - if find_name_or_class_matches(name, module, target_names) + if match_targets(name, module, target_names) )