ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update _utils.py

Browse source
This commit is contained in:
Daniel Han 2025-10-16 05:06:37 -07:00
commit bfd6b73b80
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -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