Spaces:
Runtime error
Runtime error
zetavg
commited on
Commit
•
726fa4d
1
Parent(s):
b39fdac
add finetune_params
Browse files
lora_models/alpaca-lora-7b/finetune_params.json
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"num_train_epochs": 10,
|
3 |
+
"learning_rate": 0.0003,
|
4 |
+
"cutoff_len": 512,
|
5 |
+
"lora_r": 16,
|
6 |
+
"lora_alpha": 16,
|
7 |
+
"lora_dropout": 0.05,
|
8 |
+
"lora_target_modules": [
|
9 |
+
"q_proj",
|
10 |
+
"v_proj",
|
11 |
+
"k_proj",
|
12 |
+
"o_proj"
|
13 |
+
],
|
14 |
+
"train_on_inputs": true,
|
15 |
+
"group_by_length": false,
|
16 |
+
"save_steps": 2000,
|
17 |
+
"save_total_limit": 10,
|
18 |
+
"logging_steps": 10
|
19 |
+
}
|
lora_models/unhelpful-ai-v01/finetune_params.json
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"num_train_epochs": 16,
|
3 |
+
"learning_rate": 0.0003,
|
4 |
+
"cutoff_len": 512,
|
5 |
+
"lora_r": 12,
|
6 |
+
"lora_alpha": 32,
|
7 |
+
"lora_dropout": 0.05,
|
8 |
+
"lora_target_modules": [
|
9 |
+
"q_proj",
|
10 |
+
"v_proj",
|
11 |
+
"k_proj",
|
12 |
+
"o_proj"
|
13 |
+
],
|
14 |
+
"train_on_inputs": false,
|
15 |
+
"group_by_length": false,
|
16 |
+
"save_steps": 500,
|
17 |
+
"save_total_limit": 5,
|
18 |
+
"logging_steps": 10
|
19 |
+
}
|