update model card README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,6 @@ tags:
|
|
6 |
model-index:
|
7 |
- name: empath-falcon-40b
|
8 |
results: []
|
9 |
-
library_name: peft
|
10 |
---
|
11 |
|
12 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -30,26 +29,15 @@ More information needed
|
|
30 |
|
31 |
## Training procedure
|
32 |
|
33 |
-
|
34 |
-
The following `bitsandbytes` quantization config was used during training:
|
35 |
-
- load_in_8bit: False
|
36 |
-
- load_in_4bit: True
|
37 |
-
- llm_int8_threshold: 6.0
|
38 |
-
- llm_int8_skip_modules: None
|
39 |
-
- llm_int8_enable_fp32_cpu_offload: False
|
40 |
-
- llm_int8_has_fp16_weight: False
|
41 |
-
- bnb_4bit_quant_type: nf4
|
42 |
-
- bnb_4bit_use_double_quant: True
|
43 |
-
- bnb_4bit_compute_dtype: bfloat16
|
44 |
### Training hyperparameters
|
45 |
|
46 |
The following hyperparameters were used during training:
|
47 |
- learning_rate: 0.0002
|
48 |
-
- train_batch_size:
|
49 |
- eval_batch_size: 8
|
50 |
- seed: 42
|
51 |
- gradient_accumulation_steps: 4
|
52 |
-
- total_train_batch_size:
|
53 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
54 |
- lr_scheduler_type: cosine
|
55 |
- lr_scheduler_warmup_ratio: 0.05
|
@@ -61,7 +49,6 @@ The following hyperparameters were used during training:
|
|
61 |
|
62 |
### Framework versions
|
63 |
|
64 |
-
- PEFT 0.5.0.dev0
|
65 |
- Transformers 4.32.0.dev0
|
66 |
- Pytorch 2.0.1+cu117
|
67 |
- Datasets 2.13.1
|
|
|
6 |
model-index:
|
7 |
- name: empath-falcon-40b
|
8 |
results: []
|
|
|
9 |
---
|
10 |
|
11 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
29 |
|
30 |
## Training procedure
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
### Training hyperparameters
|
33 |
|
34 |
The following hyperparameters were used during training:
|
35 |
- learning_rate: 0.0002
|
36 |
+
- train_batch_size: 4
|
37 |
- eval_batch_size: 8
|
38 |
- seed: 42
|
39 |
- gradient_accumulation_steps: 4
|
40 |
+
- total_train_batch_size: 16
|
41 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
42 |
- lr_scheduler_type: cosine
|
43 |
- lr_scheduler_warmup_ratio: 0.05
|
|
|
49 |
|
50 |
### Framework versions
|
51 |
|
|
|
52 |
- Transformers 4.32.0.dev0
|
53 |
- Pytorch 2.0.1+cu117
|
54 |
- Datasets 2.13.1
|