We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0828514 commit cfedcaeCopy full SHA for cfedcae
recipes/tests/configs/test_configs.py
@@ -7,13 +7,14 @@
7
8
import pytest
9
10
-from recipes.params import FullFinetuneParams
+from recipes.params import FullFinetuneParams, LoRAFinetuneParams
11
from torchtune.utils.argparse import TuneArgumentParser
12
13
ROOT_DIR: str = os.path.join(os.path.abspath(__file__), "../../../configs")
14
15
config_to_params = {
16
os.path.join(ROOT_DIR, "alpaca_llama2_full_finetune.yaml"): FullFinetuneParams,
17
+ os.path.join(ROOT_DIR, "alpaca_llama2_lora_finetune.yaml"): LoRAFinetuneParams,
18
}
19
20
0 commit comments