First attempt at a BitsNBytes QLoRA Falcon model.
Browse files- checkpoint-100/README.md +32 -0
- checkpoint-100/adapter_config.json +22 -0
- checkpoint-100/adapter_model.bin +3 -0
- checkpoint-100/optimizer.pt +3 -0
- checkpoint-100/rng_state.pth +3 -0
- checkpoint-100/scheduler.pt +3 -0
- checkpoint-100/special_tokens_map.json +17 -0
- checkpoint-100/tokenizer.json +0 -0
- checkpoint-100/tokenizer_config.json +7 -0
- checkpoint-100/trainer_state.json +76 -0
- checkpoint-100/training_args.bin +3 -0
- checkpoint-150/README.md +32 -0
- checkpoint-150/adapter_config.json +22 -0
- checkpoint-150/adapter_model.bin +3 -0
- checkpoint-150/optimizer.pt +3 -0
- checkpoint-150/rng_state.pth +3 -0
- checkpoint-150/scheduler.pt +3 -0
- checkpoint-150/special_tokens_map.json +17 -0
- checkpoint-150/tokenizer.json +0 -0
- checkpoint-150/tokenizer_config.json +7 -0
- checkpoint-150/trainer_state.json +106 -0
- checkpoint-150/training_args.bin +3 -0
- checkpoint-50/README.md +32 -0
- checkpoint-50/adapter_config.json +22 -0
- checkpoint-50/adapter_model.bin +3 -0
- checkpoint-50/optimizer.pt +3 -0
- checkpoint-50/rng_state.pth +3 -0
- checkpoint-50/scheduler.pt +3 -0
- checkpoint-50/special_tokens_map.json +17 -0
- checkpoint-50/tokenizer.json +0 -0
- checkpoint-50/tokenizer_config.json +7 -0
- checkpoint-50/trainer_state.json +46 -0
- checkpoint-50/training_args.bin +3 -0
checkpoint-100/README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: peft
|
| 3 |
+
---
|
| 4 |
+
## Training procedure
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 8 |
+
- load_in_8bit: False
|
| 9 |
+
- load_in_4bit: True
|
| 10 |
+
- llm_int8_threshold: 6.0
|
| 11 |
+
- llm_int8_skip_modules: None
|
| 12 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 13 |
+
- llm_int8_has_fp16_weight: False
|
| 14 |
+
- bnb_4bit_quant_type: nf4
|
| 15 |
+
- bnb_4bit_use_double_quant: False
|
| 16 |
+
- bnb_4bit_compute_dtype: float16
|
| 17 |
+
|
| 18 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 19 |
+
- load_in_8bit: False
|
| 20 |
+
- load_in_4bit: True
|
| 21 |
+
- llm_int8_threshold: 6.0
|
| 22 |
+
- llm_int8_skip_modules: None
|
| 23 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 24 |
+
- llm_int8_has_fp16_weight: False
|
| 25 |
+
- bnb_4bit_quant_type: nf4
|
| 26 |
+
- bnb_4bit_use_double_quant: False
|
| 27 |
+
- bnb_4bit_compute_dtype: float16
|
| 28 |
+
### Framework versions
|
| 29 |
+
|
| 30 |
+
- PEFT 0.4.0.dev0
|
| 31 |
+
|
| 32 |
+
- PEFT 0.4.0.dev0
|
checkpoint-100/adapter_config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"base_model_name_or_path": "ybelkada/falcon-7b-sharded-bf16",
|
| 3 |
+
"bias": "none",
|
| 4 |
+
"fan_in_fan_out": false,
|
| 5 |
+
"inference_mode": true,
|
| 6 |
+
"init_lora_weights": true,
|
| 7 |
+
"layers_pattern": null,
|
| 8 |
+
"layers_to_transform": null,
|
| 9 |
+
"lora_alpha": 16,
|
| 10 |
+
"lora_dropout": 0.1,
|
| 11 |
+
"modules_to_save": null,
|
| 12 |
+
"peft_type": "LORA",
|
| 13 |
+
"r": 64,
|
| 14 |
+
"revision": null,
|
| 15 |
+
"target_modules": [
|
| 16 |
+
"query_key_value",
|
| 17 |
+
"dense",
|
| 18 |
+
"dense_h_to_4h",
|
| 19 |
+
"dense_4h_to_h"
|
| 20 |
+
],
|
| 21 |
+
"task_type": "CAUSAL_LM"
|
| 22 |
+
}
|
checkpoint-100/adapter_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e6a68180625d7f5344f7ba7ab4ed4e819fc1acf6fc772bd9f64b1cd8586371f1
|
| 3 |
+
size 522284877
|
checkpoint-100/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cff3861facd3ab275760f6d069a4b60d4fbf40be3d0a4061f8c880627ff246d9
|
| 3 |
+
size 1044539653
|
checkpoint-100/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b55955509ba1d00714e9cce86f80ea357c7127e2be1ba37f034cfb1bb081bd85
|
| 3 |
+
size 14575
|
checkpoint-100/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:216f76b8039f833c337db298c81f13b12082d5fd4f9d866cecd34b2ca7550b37
|
| 3 |
+
size 627
|
checkpoint-100/special_tokens_map.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
">>TITLE<<",
|
| 4 |
+
">>ABSTRACT<<",
|
| 5 |
+
">>INTRODUCTION<<",
|
| 6 |
+
">>SUMMARY<<",
|
| 7 |
+
">>COMMENT<<",
|
| 8 |
+
">>ANSWER<<",
|
| 9 |
+
">>QUESTION<<",
|
| 10 |
+
">>DOMAIN<<",
|
| 11 |
+
">>PREFIX<<",
|
| 12 |
+
">>SUFFIX<<",
|
| 13 |
+
">>MIDDLE<<"
|
| 14 |
+
],
|
| 15 |
+
"eos_token": "<|endoftext|>",
|
| 16 |
+
"pad_token": "<|endoftext|>"
|
| 17 |
+
}
|
checkpoint-100/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-100/tokenizer_config.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"clean_up_tokenization_spaces": true,
|
| 4 |
+
"eos_token": "<|endoftext|>",
|
| 5 |
+
"model_max_length": 2048,
|
| 6 |
+
"tokenizer_class": "PreTrainedTokenizerFast"
|
| 7 |
+
}
|
checkpoint-100/trainer_state.json
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_metric": null,
|
| 3 |
+
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 0.64,
|
| 5 |
+
"global_step": 100,
|
| 6 |
+
"is_hyper_param_search": false,
|
| 7 |
+
"is_local_process_zero": true,
|
| 8 |
+
"is_world_process_zero": true,
|
| 9 |
+
"log_history": [
|
| 10 |
+
{
|
| 11 |
+
"epoch": 0.06,
|
| 12 |
+
"learning_rate": 0.0002,
|
| 13 |
+
"loss": 2.8722,
|
| 14 |
+
"step": 10
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"epoch": 0.13,
|
| 18 |
+
"learning_rate": 0.0002,
|
| 19 |
+
"loss": 2.7951,
|
| 20 |
+
"step": 20
|
| 21 |
+
},
|
| 22 |
+
{
|
| 23 |
+
"epoch": 0.19,
|
| 24 |
+
"learning_rate": 0.0002,
|
| 25 |
+
"loss": 2.7847,
|
| 26 |
+
"step": 30
|
| 27 |
+
},
|
| 28 |
+
{
|
| 29 |
+
"epoch": 0.26,
|
| 30 |
+
"learning_rate": 0.0002,
|
| 31 |
+
"loss": 2.7507,
|
| 32 |
+
"step": 40
|
| 33 |
+
},
|
| 34 |
+
{
|
| 35 |
+
"epoch": 0.32,
|
| 36 |
+
"learning_rate": 0.0002,
|
| 37 |
+
"loss": 2.7355,
|
| 38 |
+
"step": 50
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.38,
|
| 42 |
+
"learning_rate": 0.0002,
|
| 43 |
+
"loss": 2.7085,
|
| 44 |
+
"step": 60
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"epoch": 0.45,
|
| 48 |
+
"learning_rate": 0.0002,
|
| 49 |
+
"loss": 2.6827,
|
| 50 |
+
"step": 70
|
| 51 |
+
},
|
| 52 |
+
{
|
| 53 |
+
"epoch": 0.51,
|
| 54 |
+
"learning_rate": 0.0002,
|
| 55 |
+
"loss": 2.6429,
|
| 56 |
+
"step": 80
|
| 57 |
+
},
|
| 58 |
+
{
|
| 59 |
+
"epoch": 0.58,
|
| 60 |
+
"learning_rate": 0.0002,
|
| 61 |
+
"loss": 2.6901,
|
| 62 |
+
"step": 90
|
| 63 |
+
},
|
| 64 |
+
{
|
| 65 |
+
"epoch": 0.64,
|
| 66 |
+
"learning_rate": 0.0002,
|
| 67 |
+
"loss": 2.688,
|
| 68 |
+
"step": 100
|
| 69 |
+
}
|
| 70 |
+
],
|
| 71 |
+
"max_steps": 156,
|
| 72 |
+
"num_train_epochs": 1,
|
| 73 |
+
"total_flos": 6.770832332292096e+16,
|
| 74 |
+
"trial_name": null,
|
| 75 |
+
"trial_params": null
|
| 76 |
+
}
|
checkpoint-100/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06aa948ab085088ada56ecf2747c82338501c15fcdff5b592094917bcb713d9d
|
| 3 |
+
size 3963
|
checkpoint-150/README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: peft
|
| 3 |
+
---
|
| 4 |
+
## Training procedure
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 8 |
+
- load_in_8bit: False
|
| 9 |
+
- load_in_4bit: True
|
| 10 |
+
- llm_int8_threshold: 6.0
|
| 11 |
+
- llm_int8_skip_modules: None
|
| 12 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 13 |
+
- llm_int8_has_fp16_weight: False
|
| 14 |
+
- bnb_4bit_quant_type: nf4
|
| 15 |
+
- bnb_4bit_use_double_quant: False
|
| 16 |
+
- bnb_4bit_compute_dtype: float16
|
| 17 |
+
|
| 18 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 19 |
+
- load_in_8bit: False
|
| 20 |
+
- load_in_4bit: True
|
| 21 |
+
- llm_int8_threshold: 6.0
|
| 22 |
+
- llm_int8_skip_modules: None
|
| 23 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 24 |
+
- llm_int8_has_fp16_weight: False
|
| 25 |
+
- bnb_4bit_quant_type: nf4
|
| 26 |
+
- bnb_4bit_use_double_quant: False
|
| 27 |
+
- bnb_4bit_compute_dtype: float16
|
| 28 |
+
### Framework versions
|
| 29 |
+
|
| 30 |
+
- PEFT 0.4.0.dev0
|
| 31 |
+
|
| 32 |
+
- PEFT 0.4.0.dev0
|
checkpoint-150/adapter_config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"base_model_name_or_path": "ybelkada/falcon-7b-sharded-bf16",
|
| 3 |
+
"bias": "none",
|
| 4 |
+
"fan_in_fan_out": false,
|
| 5 |
+
"inference_mode": true,
|
| 6 |
+
"init_lora_weights": true,
|
| 7 |
+
"layers_pattern": null,
|
| 8 |
+
"layers_to_transform": null,
|
| 9 |
+
"lora_alpha": 16,
|
| 10 |
+
"lora_dropout": 0.1,
|
| 11 |
+
"modules_to_save": null,
|
| 12 |
+
"peft_type": "LORA",
|
| 13 |
+
"r": 64,
|
| 14 |
+
"revision": null,
|
| 15 |
+
"target_modules": [
|
| 16 |
+
"query_key_value",
|
| 17 |
+
"dense",
|
| 18 |
+
"dense_h_to_4h",
|
| 19 |
+
"dense_4h_to_h"
|
| 20 |
+
],
|
| 21 |
+
"task_type": "CAUSAL_LM"
|
| 22 |
+
}
|
checkpoint-150/adapter_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:76ee304e50331fa3057446aefbc1ca2d020137066725f7c3b4dc74cfd8b97782
|
| 3 |
+
size 522284877
|
checkpoint-150/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:00ba98f1157ac1c94ca39537937548e4937574f90f4b8ebbf46d054b4e75aff5
|
| 3 |
+
size 1044539653
|
checkpoint-150/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5005f87b4048f2a98b8e173effa7b158007e428435416ebe049c79042f57bec3
|
| 3 |
+
size 14575
|
checkpoint-150/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:01422943191c18f4ce94fe70c7ffd85d7e6e08513a418ac2779ae0b3b3ba38a4
|
| 3 |
+
size 627
|
checkpoint-150/special_tokens_map.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
">>TITLE<<",
|
| 4 |
+
">>ABSTRACT<<",
|
| 5 |
+
">>INTRODUCTION<<",
|
| 6 |
+
">>SUMMARY<<",
|
| 7 |
+
">>COMMENT<<",
|
| 8 |
+
">>ANSWER<<",
|
| 9 |
+
">>QUESTION<<",
|
| 10 |
+
">>DOMAIN<<",
|
| 11 |
+
">>PREFIX<<",
|
| 12 |
+
">>SUFFIX<<",
|
| 13 |
+
">>MIDDLE<<"
|
| 14 |
+
],
|
| 15 |
+
"eos_token": "<|endoftext|>",
|
| 16 |
+
"pad_token": "<|endoftext|>"
|
| 17 |
+
}
|
checkpoint-150/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-150/tokenizer_config.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"clean_up_tokenization_spaces": true,
|
| 4 |
+
"eos_token": "<|endoftext|>",
|
| 5 |
+
"model_max_length": 2048,
|
| 6 |
+
"tokenizer_class": "PreTrainedTokenizerFast"
|
| 7 |
+
}
|
checkpoint-150/trainer_state.json
ADDED
|
@@ -0,0 +1,106 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_metric": null,
|
| 3 |
+
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 0.96,
|
| 5 |
+
"global_step": 150,
|
| 6 |
+
"is_hyper_param_search": false,
|
| 7 |
+
"is_local_process_zero": true,
|
| 8 |
+
"is_world_process_zero": true,
|
| 9 |
+
"log_history": [
|
| 10 |
+
{
|
| 11 |
+
"epoch": 0.06,
|
| 12 |
+
"learning_rate": 0.0002,
|
| 13 |
+
"loss": 2.8722,
|
| 14 |
+
"step": 10
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"epoch": 0.13,
|
| 18 |
+
"learning_rate": 0.0002,
|
| 19 |
+
"loss": 2.7951,
|
| 20 |
+
"step": 20
|
| 21 |
+
},
|
| 22 |
+
{
|
| 23 |
+
"epoch": 0.19,
|
| 24 |
+
"learning_rate": 0.0002,
|
| 25 |
+
"loss": 2.7847,
|
| 26 |
+
"step": 30
|
| 27 |
+
},
|
| 28 |
+
{
|
| 29 |
+
"epoch": 0.26,
|
| 30 |
+
"learning_rate": 0.0002,
|
| 31 |
+
"loss": 2.7507,
|
| 32 |
+
"step": 40
|
| 33 |
+
},
|
| 34 |
+
{
|
| 35 |
+
"epoch": 0.32,
|
| 36 |
+
"learning_rate": 0.0002,
|
| 37 |
+
"loss": 2.7355,
|
| 38 |
+
"step": 50
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.38,
|
| 42 |
+
"learning_rate": 0.0002,
|
| 43 |
+
"loss": 2.7085,
|
| 44 |
+
"step": 60
|
| 45 |
+
},
|
| 46 |
+
{
|
| 47 |
+
"epoch": 0.45,
|
| 48 |
+
"learning_rate": 0.0002,
|
| 49 |
+
"loss": 2.6827,
|
| 50 |
+
"step": 70
|
| 51 |
+
},
|
| 52 |
+
{
|
| 53 |
+
"epoch": 0.51,
|
| 54 |
+
"learning_rate": 0.0002,
|
| 55 |
+
"loss": 2.6429,
|
| 56 |
+
"step": 80
|
| 57 |
+
},
|
| 58 |
+
{
|
| 59 |
+
"epoch": 0.58,
|
| 60 |
+
"learning_rate": 0.0002,
|
| 61 |
+
"loss": 2.6901,
|
| 62 |
+
"step": 90
|
| 63 |
+
},
|
| 64 |
+
{
|
| 65 |
+
"epoch": 0.64,
|
| 66 |
+
"learning_rate": 0.0002,
|
| 67 |
+
"loss": 2.688,
|
| 68 |
+
"step": 100
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"epoch": 0.7,
|
| 72 |
+
"learning_rate": 0.0002,
|
| 73 |
+
"loss": 2.7024,
|
| 74 |
+
"step": 110
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"epoch": 0.77,
|
| 78 |
+
"learning_rate": 0.0002,
|
| 79 |
+
"loss": 2.6776,
|
| 80 |
+
"step": 120
|
| 81 |
+
},
|
| 82 |
+
{
|
| 83 |
+
"epoch": 0.83,
|
| 84 |
+
"learning_rate": 0.0002,
|
| 85 |
+
"loss": 2.6491,
|
| 86 |
+
"step": 130
|
| 87 |
+
},
|
| 88 |
+
{
|
| 89 |
+
"epoch": 0.9,
|
| 90 |
+
"learning_rate": 0.0002,
|
| 91 |
+
"loss": 2.6498,
|
| 92 |
+
"step": 140
|
| 93 |
+
},
|
| 94 |
+
{
|
| 95 |
+
"epoch": 0.96,
|
| 96 |
+
"learning_rate": 0.0002,
|
| 97 |
+
"loss": 2.6607,
|
| 98 |
+
"step": 150
|
| 99 |
+
}
|
| 100 |
+
],
|
| 101 |
+
"max_steps": 156,
|
| 102 |
+
"num_train_epochs": 1,
|
| 103 |
+
"total_flos": 1.0156248498438144e+17,
|
| 104 |
+
"trial_name": null,
|
| 105 |
+
"trial_params": null
|
| 106 |
+
}
|
checkpoint-150/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06aa948ab085088ada56ecf2747c82338501c15fcdff5b592094917bcb713d9d
|
| 3 |
+
size 3963
|
checkpoint-50/README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: peft
|
| 3 |
+
---
|
| 4 |
+
## Training procedure
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 8 |
+
- load_in_8bit: False
|
| 9 |
+
- load_in_4bit: True
|
| 10 |
+
- llm_int8_threshold: 6.0
|
| 11 |
+
- llm_int8_skip_modules: None
|
| 12 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 13 |
+
- llm_int8_has_fp16_weight: False
|
| 14 |
+
- bnb_4bit_quant_type: nf4
|
| 15 |
+
- bnb_4bit_use_double_quant: False
|
| 16 |
+
- bnb_4bit_compute_dtype: float16
|
| 17 |
+
|
| 18 |
+
The following `bitsandbytes` quantization config was used during training:
|
| 19 |
+
- load_in_8bit: False
|
| 20 |
+
- load_in_4bit: True
|
| 21 |
+
- llm_int8_threshold: 6.0
|
| 22 |
+
- llm_int8_skip_modules: None
|
| 23 |
+
- llm_int8_enable_fp32_cpu_offload: False
|
| 24 |
+
- llm_int8_has_fp16_weight: False
|
| 25 |
+
- bnb_4bit_quant_type: nf4
|
| 26 |
+
- bnb_4bit_use_double_quant: False
|
| 27 |
+
- bnb_4bit_compute_dtype: float16
|
| 28 |
+
### Framework versions
|
| 29 |
+
|
| 30 |
+
- PEFT 0.4.0.dev0
|
| 31 |
+
|
| 32 |
+
- PEFT 0.4.0.dev0
|
checkpoint-50/adapter_config.json
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"base_model_name_or_path": "ybelkada/falcon-7b-sharded-bf16",
|
| 3 |
+
"bias": "none",
|
| 4 |
+
"fan_in_fan_out": false,
|
| 5 |
+
"inference_mode": true,
|
| 6 |
+
"init_lora_weights": true,
|
| 7 |
+
"layers_pattern": null,
|
| 8 |
+
"layers_to_transform": null,
|
| 9 |
+
"lora_alpha": 16,
|
| 10 |
+
"lora_dropout": 0.1,
|
| 11 |
+
"modules_to_save": null,
|
| 12 |
+
"peft_type": "LORA",
|
| 13 |
+
"r": 64,
|
| 14 |
+
"revision": null,
|
| 15 |
+
"target_modules": [
|
| 16 |
+
"query_key_value",
|
| 17 |
+
"dense",
|
| 18 |
+
"dense_h_to_4h",
|
| 19 |
+
"dense_4h_to_h"
|
| 20 |
+
],
|
| 21 |
+
"task_type": "CAUSAL_LM"
|
| 22 |
+
}
|
checkpoint-50/adapter_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:032f42d9d5c258856e19effca1ecaee7ec61f7188a473342290595c436c2adff
|
| 3 |
+
size 522284877
|
checkpoint-50/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39889cd8f1fad5b588fd13c98ed1b4c4284f8927f417f4bdb700da6e9ba3ee7a
|
| 3 |
+
size 1044539653
|
checkpoint-50/rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:44ad36cf584d2de00c5938d038a775667aba2a65d3fa5d28d0ef7b3ba129f0d7
|
| 3 |
+
size 14575
|
checkpoint-50/scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d217edb059555be97837810de03dc22d79976c8832aac609d084f6016b5003c4
|
| 3 |
+
size 627
|
checkpoint-50/special_tokens_map.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
">>TITLE<<",
|
| 4 |
+
">>ABSTRACT<<",
|
| 5 |
+
">>INTRODUCTION<<",
|
| 6 |
+
">>SUMMARY<<",
|
| 7 |
+
">>COMMENT<<",
|
| 8 |
+
">>ANSWER<<",
|
| 9 |
+
">>QUESTION<<",
|
| 10 |
+
">>DOMAIN<<",
|
| 11 |
+
">>PREFIX<<",
|
| 12 |
+
">>SUFFIX<<",
|
| 13 |
+
">>MIDDLE<<"
|
| 14 |
+
],
|
| 15 |
+
"eos_token": "<|endoftext|>",
|
| 16 |
+
"pad_token": "<|endoftext|>"
|
| 17 |
+
}
|
checkpoint-50/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint-50/tokenizer_config.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"clean_up_tokenization_spaces": true,
|
| 4 |
+
"eos_token": "<|endoftext|>",
|
| 5 |
+
"model_max_length": 2048,
|
| 6 |
+
"tokenizer_class": "PreTrainedTokenizerFast"
|
| 7 |
+
}
|
checkpoint-50/trainer_state.json
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_metric": null,
|
| 3 |
+
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 0.32,
|
| 5 |
+
"global_step": 50,
|
| 6 |
+
"is_hyper_param_search": false,
|
| 7 |
+
"is_local_process_zero": true,
|
| 8 |
+
"is_world_process_zero": true,
|
| 9 |
+
"log_history": [
|
| 10 |
+
{
|
| 11 |
+
"epoch": 0.06,
|
| 12 |
+
"learning_rate": 0.0002,
|
| 13 |
+
"loss": 2.8722,
|
| 14 |
+
"step": 10
|
| 15 |
+
},
|
| 16 |
+
{
|
| 17 |
+
"epoch": 0.13,
|
| 18 |
+
"learning_rate": 0.0002,
|
| 19 |
+
"loss": 2.7951,
|
| 20 |
+
"step": 20
|
| 21 |
+
},
|
| 22 |
+
{
|
| 23 |
+
"epoch": 0.19,
|
| 24 |
+
"learning_rate": 0.0002,
|
| 25 |
+
"loss": 2.7847,
|
| 26 |
+
"step": 30
|
| 27 |
+
},
|
| 28 |
+
{
|
| 29 |
+
"epoch": 0.26,
|
| 30 |
+
"learning_rate": 0.0002,
|
| 31 |
+
"loss": 2.7507,
|
| 32 |
+
"step": 40
|
| 33 |
+
},
|
| 34 |
+
{
|
| 35 |
+
"epoch": 0.32,
|
| 36 |
+
"learning_rate": 0.0002,
|
| 37 |
+
"loss": 2.7355,
|
| 38 |
+
"step": 50
|
| 39 |
+
}
|
| 40 |
+
],
|
| 41 |
+
"max_steps": 156,
|
| 42 |
+
"num_train_epochs": 1,
|
| 43 |
+
"total_flos": 3.385416166146048e+16,
|
| 44 |
+
"trial_name": null,
|
| 45 |
+
"trial_params": null
|
| 46 |
+
}
|
checkpoint-50/training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06aa948ab085088ada56ecf2747c82338501c15fcdff5b592094917bcb713d9d
|
| 3 |
+
size 3963
|