update test

Lightning-AI · Mar 6, 2024 · 6b539c4 · 6b539c4
1 parent e108211
commit 6b539c4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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."""