Format

neuralmagic · Jul 18, 2024 · 00a8cd8 · 00a8cd8
1 parent ba7d420
commit 00a8cd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)}")
+        print(f"Saved final checkpoint to {os.path.abspath(save_directory)}")