From 1dd4df999296dea3a2ea1e150fb73c4cb4d56412 Mon Sep 17 00:00:00 2001 From: Pavel T Date: Fri, 23 May 2025 14:30:35 -0400 Subject: [PATCH] fix _replace_with_custom_fn_if_matches_filter in quant_api.py so it calls `filter_fn` according to definition --- torchao/quantization/quant_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchao/quantization/quant_api.py b/torchao/quantization/quant_api.py index 4229577b95..a657f602c1 100644 --- a/torchao/quantization/quant_api.py +++ b/torchao/quantization/quant_api.py @@ -285,7 +285,7 @@ def _replace_with_custom_fn_if_matches_filter( new_module.weight = model.weight new_module.bias = model.bias model = new_module - if filter_fn(model, cur_fqn[:-1]): + if filter_fn(model): if device is not None: model.to(device=device) # move to device before quantization model = replacement_fn(model, *extra_args)