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

Update save.py

Browse source
This commit is contained in:
Daniel Han 2025-12-17 02:28:03 -08:00
commit 4063b87c01
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/save.py
View file

@ -2817,7 +2817,10 @@ def _unsloth_save_torchao_with_given_config(
)
from torchao import quantize_
quantization_config = TorchAoConfig(quant_type = torchao_config)
if isinstance(torchao_config, TorchAoConfig):
quantization_config = torchao_config
else:
quantization_config = TorchAoConfig(quant_type = torchao_config)
# Determine if this is a VLM
is_vlm = False