diff --git a/tests/test_config_hub.py b/tests/test_config_hub.py index b0aa4daf3f..0eb6a7a476 100644 --- a/tests/test_config_hub.py +++ b/tests/test_config_hub.py @@ -11,6 +11,9 @@ ("lit_gpt/pretrain.py", "pretrain/debug.yaml"), ("lit_gpt/pretrain.py", "pretrain/tinyllama.yaml"), ("lit_gpt/pretrain.py", "pretrain/tinystories.yaml"), + ("finetune/full.py", "finetune/llama-2-7b/full.yaml"), + ("finetune/lora.py", "finetune/llama-2-7b/lora.yaml"), + ("finetune/lora.py", "finetune/tiny-llama/lora.yaml"), ]) def test_config_help(script_file, config_file, monkeypatch, tmp_path): """Test that configs validate against the signature in the scripts."""