black

bytelatent/checkpoint.py

@@ -325,15 +325,19 @@ class CheckpointManager:
 
         return cls(args)
 
+
 def main(
     command: str,
     model_checkpoint_dir: str,
 ):
     if command == "consolidate":
-        print(f"Consolidating {model_checkpoint_dir}. Output will be in the {CONSOLIDATE_FOLDER} folder.")
+        print(
+            f"Consolidating {model_checkpoint_dir}. Output will be in the {CONSOLIDATE_FOLDER} folder."
+        )
         consolidate_checkpoints(fsspec.filesystem("file"), model_checkpoint_dir)
     else:
         raise ValueError("Invalid command")
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     typer.run(main)