We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dd7f19d commit 4837b68Copy full SHA for 4837b68
src/lightning/fabric/cli.py
@@ -60,6 +60,7 @@ def _get_supported_strategies() -> list[str]:
60
61
set_config_read_mode(fsspec_enabled=True)
62
63
+
64
class FabricCLI:
65
"""Lightning Fabric command-line tool."""
66
@@ -196,6 +197,7 @@ def _consolidate_checkpoint(self) -> None:
196
197
checkpoint = _load_distributed_checkpoint(processed_args.checkpoint_folder)
198
torch.save(checkpoint, processed_args.output_file)
199
200
201
def _entrypoint() -> None:
202
"""The CLI entrypoint."""
203
FabricCLI()
0 commit comments