tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

prevent cli functions from getting fired on import (#581)

Browse Source
This commit is contained in:
Wing Lian
2023-09-15 04:03:32 -04:00
committed by GitHub
parent a5a625f47e
commit 8dcd40ac78
4 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/cli/inference.py
View File

@@ -23,4 +23,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
do_inference(cfg=parsed_cfg, cli_args=parsed_cli_args)
fire.Fire(do_cli)
if __name__ == "__main__":
fire.Fire(do_cli)

3
src/axolotl/cli/merge_lora.py
View File

@@ -23,4 +23,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
do_merge_lora(cfg=parsed_cfg, cli_args=parsed_cli_args)
fire.Fire(do_cli)
if __name__ == "__main__":
fire.Fire(do_cli)

3
src/axolotl/cli/shard.py
View File

@@ -38,4 +38,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
shard(cfg=parsed_cfg, cli_args=parsed_cli_args)
fire.Fire(do_cli)
if __name__ == "__main__":
fire.Fire(do_cli)

3
src/axolotl/cli/train.py
View File

@@ -32,4 +32,5 @@ def do_cli(config: Path = Path("examples/"), **kwargs):
train(cfg=parsed_cfg, cli_args=parsed_cli_args, dataset_meta=dataset_meta)
fire.Fire(do_cli)
if __name__ == "__main__":
fire.Fire(do_cli)