From bfd6b73b80746f70cb7e320b442a1d2f43c01fd9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 16 Oct 2025 05:06:37 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index eb5cc28a2b..6728044c8e 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1683,7 +1683,7 @@ def _prepare_model_for_qat(model: torch.nn.Module, qat_scheme: Union[str, TorchA from torchao.quantization.granularity import PerGroup from torchao.quantization.qat import QATConfig - if isinstance(qat_scheme, TorchAOConfig): + if not isinstance(qat_scheme, TorchAOConfig): filter_fn = None group_size = None base_config = None