diff --git a/src/axolotl/cli/inference.py b/src/axolotl/cli/inference.py index 1a5a1a2be..f3daac83d 100644 --- a/src/axolotl/cli/inference.py +++ b/src/axolotl/cli/inference.py @@ -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) diff --git a/src/axolotl/cli/merge_lora.py b/src/axolotl/cli/merge_lora.py index 473aa8260..79b7112b5 100644 --- a/src/axolotl/cli/merge_lora.py +++ b/src/axolotl/cli/merge_lora.py @@ -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) diff --git a/src/axolotl/cli/shard.py b/src/axolotl/cli/shard.py index ad7d9a136..85901b0f2 100644 --- a/src/axolotl/cli/shard.py +++ b/src/axolotl/cli/shard.py @@ -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) diff --git a/src/axolotl/cli/train.py b/src/axolotl/cli/train.py index 166af2595..72a9250c8 100644 --- a/src/axolotl/cli/train.py +++ b/src/axolotl/cli/train.py @@ -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)