Skip to content

Commit 1d812a0

Browse files
xin3heKaihui-intel
andauthored
Update neural_compressor/torch/algorithms/weight_only/autoround.py
Co-authored-by: Tang Kaihui <kaihui.tang@intel.com>
1 parent bca2063 commit 1d812a0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

neural_compressor/torch/algorithms/weight_only/autoround.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,7 @@ def convert(self, model: torch.nn.Module, *args, **kwargs):
377377
try:
378378
del model
379379
self.accelerator.empty_cache()
380-
logger.info("Quantization is done, reloading model from saved directory...")
380+
logger.info(f"Quantization is done, reloading model from saved directory({self.output_dir})...")
381381
import transformers # pylint: disable=E0401
382382

383383
model = transformers.AutoModelForCausalLM.from_pretrained(self.output_dir)

0 commit comments

Comments
 (0)