Format

Author: mgoin

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)}")