diff --git a/auto_fp8/modeling.py b/auto_fp8/modeling.py index 43b4d86..3fa8e75 100644 --- a/auto_fp8/modeling.py +++ b/auto_fp8/modeling.py @@ -77,4 +77,4 @@ def save_pretrained(self, save_directory: str, save_compressed: bool = True): self.model.save_pretrained(save_directory, save_compressed=save_compressed) tokenizer = AutoTokenizer.from_pretrained(self.model.config._name_or_path) tokenizer.save_pretrained(save_directory) - print(f"Saved final checkpoint to {os.path.abspath(save_directory)}") \ No newline at end of file + print(f"Saved final checkpoint to {os.path.abspath(save_directory)}")