vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-09-01 10:09:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

black

Browse source
This commit is contained in:
Srini Iyer 2025-03-11 23:51:23 +00:00
parent 699d0be470
commit 6160e83d41
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
bytelatent/checkpoint.py
View file

@ -325,15 +325,19 @@ class CheckpointManager:
return cls(args) return cls(args)
def main( def main(
command: str, command: str,
model_checkpoint_dir: str, model_checkpoint_dir: str,
): ):
if command == "consolidate": 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) consolidate_checkpoints(fsspec.filesystem("file"), model_checkpoint_dir)
else: else:
raise ValueError("Invalid command") raise ValueError("Invalid command")
if __name__ == '__main__':
if __name__ == "__main__":
typer.run(main) typer.run(main)