Skip to content

Commit cfedcae

Browse files
authored
Add LoRA recipe config to test_configs.py (#353)
1 parent 0828514 commit cfedcae

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

recipes/tests/configs/test_configs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,14 @@
77

88
import pytest
99

10-
from recipes.params import FullFinetuneParams
10+
from recipes.params import FullFinetuneParams, LoRAFinetuneParams
1111
from torchtune.utils.argparse import TuneArgumentParser
1212

1313
ROOT_DIR: str = os.path.join(os.path.abspath(__file__), "../../../configs")
1414

1515
config_to_params = {
1616
os.path.join(ROOT_DIR, "alpaca_llama2_full_finetune.yaml"): FullFinetuneParams,
17+
os.path.join(ROOT_DIR, "alpaca_llama2_lora_finetune.yaml"): LoRAFinetuneParams,
1718
}
1819

1920

0 commit comments

Comments
 (0)