Skip to content

Commit e25788e

Browse files
Added more model config parameters for qlora
1 parent ee3fc6b commit e25788e

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/kfto/core/config_qlora.json

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,13 @@
1515
"response_template": "\n### Label:",
1616
"dataset_text_field": "output",
1717
"use_flash_attn": false,
18-
"peft_method": "qlora",
18+
"peft_method": "lora",
19+
"r": 8,
20+
"lora_dropout": 0.05,
21+
"lora_alpha": 16,
1922
"target_modules": ["all-linear"],
2023
"use_4bit": true,
2124
"bnb_4bit_compute_dtype": "float16",
2225
"bnb_4bit_quant_type": "nf4",
23-
"use_nested_quant": false,
24-
"fp16": false,
25-
"bf16": false
26+
"fp16": true
2627
}

0 commit comments

Comments
 (0)