Skip to content

Commit 4ec50d1

Browse files
authored
Enhance tokenizer saving (#1057)
1 parent c5b1c41 commit 4ec50d1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

auto_round/compressors/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3006,7 +3006,7 @@ def save_quantized(
30063006
if format == "fake" or format == "qdq": # TODO fix act quantization later
30073007
self.model = self.model.to("cpu")
30083008
self.model.save_pretrained(output_dir)
3009-
if self.tokenizer is not None:
3009+
if self.tokenizer is not None and hasattr(self.tokenizer, "save_pretrained"):
30103010
self.tokenizer.save_pretrained(output_dir)
30113011
processor = kwargs.get("processor", None)
30123012
if processor is not None:

0 commit comments

Comments
 (0)