diff --git a/litgpt/utils.py b/litgpt/utils.py index 6eb7efbff4..18aea56d64 100644 --- a/litgpt/utils.py +++ b/litgpt/utils.py @@ -444,6 +444,7 @@ def save_hyperparameters(function: callable, checkpoint_dir: Path) -> None: ("finetune", "lora"), ("finetune", "adapter"), ("finetune", "adapter_v2"), + ("finetune",), ("pretrain",), ] for known_command in known_commands: diff --git a/tests/test_utils.py b/tests/test_utils.py index 9770bf98e7..554929c77f 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -267,6 +267,7 @@ def _test_function2(out_dir: Path, foo: bool = False, bar: int = 1): "command", [ "any.py", + "litgpt finetune", "litgpt finetune full", "litgpt finetune lora", "litgpt finetune adapter",