Upload folder using huggingface_hub
Browse files- config.json +35 -0
- generation_config.json +11 -0
- model-00001-of-00004.safetensors +3 -0
- model-00002-of-00004.safetensors +3 -0
- model-00003-of-00004.safetensors +3 -0
- model-00004-of-00004.safetensors +3 -0
- model.safetensors.index.json +202 -0
- optimizer.pt +3 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- special_tokens_map.json +24 -0
- tokenizer.json +0 -0
- tokenizer_config.json +130 -0
- trainer_state.json +543 -0
- training_args.bin +3 -0
config.json
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "microsoft/Phi-3-mini-4k-instruct",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"Phi3ForCausalLM"
|
| 5 |
+
],
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"auto_map": {
|
| 8 |
+
"AutoConfig": "microsoft/Phi-3-mini-4k-instruct--configuration_phi3.Phi3Config",
|
| 9 |
+
"AutoModelForCausalLM": "microsoft/Phi-3-mini-4k-instruct--modeling_phi3.Phi3ForCausalLM"
|
| 10 |
+
},
|
| 11 |
+
"bos_token_id": 1,
|
| 12 |
+
"embd_pdrop": 0.0,
|
| 13 |
+
"eos_token_id": 32000,
|
| 14 |
+
"hidden_act": "silu",
|
| 15 |
+
"hidden_size": 3072,
|
| 16 |
+
"initializer_range": 0.02,
|
| 17 |
+
"intermediate_size": 8192,
|
| 18 |
+
"max_position_embeddings": 4096,
|
| 19 |
+
"model_type": "phi3",
|
| 20 |
+
"num_attention_heads": 32,
|
| 21 |
+
"num_hidden_layers": 32,
|
| 22 |
+
"num_key_value_heads": 32,
|
| 23 |
+
"original_max_position_embeddings": 4096,
|
| 24 |
+
"pad_token_id": 32000,
|
| 25 |
+
"resid_pdrop": 0.0,
|
| 26 |
+
"rms_norm_eps": 1e-05,
|
| 27 |
+
"rope_scaling": null,
|
| 28 |
+
"rope_theta": 10000.0,
|
| 29 |
+
"sliding_window": 2047,
|
| 30 |
+
"tie_word_embeddings": false,
|
| 31 |
+
"torch_dtype": "float32",
|
| 32 |
+
"transformers_version": "4.40.0",
|
| 33 |
+
"use_cache": true,
|
| 34 |
+
"vocab_size": 32011
|
| 35 |
+
}
|
generation_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 1,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
32000,
|
| 6 |
+
32001,
|
| 7 |
+
32007
|
| 8 |
+
],
|
| 9 |
+
"pad_token_id": 32000,
|
| 10 |
+
"transformers_version": "4.40.0"
|
| 11 |
+
}
|
model-00001-of-00004.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fc0b4e52c279bb0c001bc5add4747bcd0f76ae3021e0c494b26e3bcc1b463d89
|
| 3 |
+
size 4961201152
|
model-00002-of-00004.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b85b407ef8a977cf237d09d1eea6f45cac1b034c284ad69e9a85ac53fe19fa2
|
| 3 |
+
size 4983111176
|
model-00003-of-00004.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:91c575e9af34b468803e85e571bec25711481005ef300f1de2b70ce5e4cd4e3e
|
| 3 |
+
size 4945374704
|
model-00004-of-00004.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:039a923da7450d1953c64df560a4c8289ec43bcee6490e256dcc059c66255f3f
|
| 3 |
+
size 393351296
|
model.safetensors.index.json
ADDED
|
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"metadata": {
|
| 3 |
+
"total_size": 15283015680
|
| 4 |
+
},
|
| 5 |
+
"weight_map": {
|
| 6 |
+
"lm_head.weight": "model-00004-of-00004.safetensors",
|
| 7 |
+
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
|
| 8 |
+
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 9 |
+
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 10 |
+
"model.layers.0.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 11 |
+
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 12 |
+
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 13 |
+
"model.layers.0.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 14 |
+
"model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 15 |
+
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 16 |
+
"model.layers.1.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 17 |
+
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 18 |
+
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 19 |
+
"model.layers.1.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 20 |
+
"model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 21 |
+
"model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 22 |
+
"model.layers.10.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 23 |
+
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 24 |
+
"model.layers.10.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 25 |
+
"model.layers.10.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 26 |
+
"model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 27 |
+
"model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 28 |
+
"model.layers.11.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 29 |
+
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 30 |
+
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 31 |
+
"model.layers.11.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 32 |
+
"model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 33 |
+
"model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 34 |
+
"model.layers.12.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 35 |
+
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 36 |
+
"model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 37 |
+
"model.layers.12.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 38 |
+
"model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 39 |
+
"model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 40 |
+
"model.layers.13.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 41 |
+
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 42 |
+
"model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 43 |
+
"model.layers.13.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 44 |
+
"model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 45 |
+
"model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 46 |
+
"model.layers.14.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 47 |
+
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 48 |
+
"model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 49 |
+
"model.layers.14.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 50 |
+
"model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 51 |
+
"model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 52 |
+
"model.layers.15.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 53 |
+
"model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 54 |
+
"model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 55 |
+
"model.layers.15.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 56 |
+
"model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 57 |
+
"model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 58 |
+
"model.layers.16.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 59 |
+
"model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 60 |
+
"model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 61 |
+
"model.layers.16.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 62 |
+
"model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 63 |
+
"model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 64 |
+
"model.layers.17.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 65 |
+
"model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 66 |
+
"model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 67 |
+
"model.layers.17.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 68 |
+
"model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 69 |
+
"model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 70 |
+
"model.layers.18.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 71 |
+
"model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 72 |
+
"model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 73 |
+
"model.layers.18.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 74 |
+
"model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 75 |
+
"model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 76 |
+
"model.layers.19.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 77 |
+
"model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 78 |
+
"model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 79 |
+
"model.layers.19.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 80 |
+
"model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 81 |
+
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 82 |
+
"model.layers.2.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 83 |
+
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 84 |
+
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 85 |
+
"model.layers.2.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 86 |
+
"model.layers.20.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 87 |
+
"model.layers.20.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
| 88 |
+
"model.layers.20.mlp.gate_up_proj.weight": "model-00002-of-00004.safetensors",
|
| 89 |
+
"model.layers.20.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
| 90 |
+
"model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 91 |
+
"model.layers.20.self_attn.qkv_proj.weight": "model-00002-of-00004.safetensors",
|
| 92 |
+
"model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 93 |
+
"model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 94 |
+
"model.layers.21.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 95 |
+
"model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 96 |
+
"model.layers.21.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
| 97 |
+
"model.layers.21.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 98 |
+
"model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 99 |
+
"model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 100 |
+
"model.layers.22.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 101 |
+
"model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 102 |
+
"model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 103 |
+
"model.layers.22.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 104 |
+
"model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 105 |
+
"model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 106 |
+
"model.layers.23.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 107 |
+
"model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 108 |
+
"model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 109 |
+
"model.layers.23.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 110 |
+
"model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 111 |
+
"model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 112 |
+
"model.layers.24.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 113 |
+
"model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 114 |
+
"model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 115 |
+
"model.layers.24.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 116 |
+
"model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 117 |
+
"model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 118 |
+
"model.layers.25.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 119 |
+
"model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 120 |
+
"model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 121 |
+
"model.layers.25.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 122 |
+
"model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 123 |
+
"model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 124 |
+
"model.layers.26.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 125 |
+
"model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 126 |
+
"model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 127 |
+
"model.layers.26.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 128 |
+
"model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 129 |
+
"model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 130 |
+
"model.layers.27.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 131 |
+
"model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 132 |
+
"model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 133 |
+
"model.layers.27.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 134 |
+
"model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 135 |
+
"model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 136 |
+
"model.layers.28.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 137 |
+
"model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 138 |
+
"model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 139 |
+
"model.layers.28.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 140 |
+
"model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 141 |
+
"model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 142 |
+
"model.layers.29.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 143 |
+
"model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 144 |
+
"model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 145 |
+
"model.layers.29.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 146 |
+
"model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 147 |
+
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 148 |
+
"model.layers.3.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 149 |
+
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 150 |
+
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 151 |
+
"model.layers.3.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 152 |
+
"model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 153 |
+
"model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 154 |
+
"model.layers.30.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 155 |
+
"model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 156 |
+
"model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 157 |
+
"model.layers.30.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 158 |
+
"model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 159 |
+
"model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
| 160 |
+
"model.layers.31.mlp.gate_up_proj.weight": "model-00003-of-00004.safetensors",
|
| 161 |
+
"model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
| 162 |
+
"model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
| 163 |
+
"model.layers.31.self_attn.qkv_proj.weight": "model-00003-of-00004.safetensors",
|
| 164 |
+
"model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 165 |
+
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 166 |
+
"model.layers.4.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 167 |
+
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 168 |
+
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 169 |
+
"model.layers.4.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 170 |
+
"model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 171 |
+
"model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 172 |
+
"model.layers.5.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 173 |
+
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 174 |
+
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 175 |
+
"model.layers.5.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 176 |
+
"model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 177 |
+
"model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 178 |
+
"model.layers.6.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 179 |
+
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 180 |
+
"model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 181 |
+
"model.layers.6.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 182 |
+
"model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 183 |
+
"model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 184 |
+
"model.layers.7.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 185 |
+
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 186 |
+
"model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 187 |
+
"model.layers.7.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 188 |
+
"model.layers.8.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 189 |
+
"model.layers.8.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 190 |
+
"model.layers.8.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 191 |
+
"model.layers.8.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 192 |
+
"model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 193 |
+
"model.layers.8.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 194 |
+
"model.layers.9.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 195 |
+
"model.layers.9.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
| 196 |
+
"model.layers.9.mlp.gate_up_proj.weight": "model-00001-of-00004.safetensors",
|
| 197 |
+
"model.layers.9.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
| 198 |
+
"model.layers.9.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
| 199 |
+
"model.layers.9.self_attn.qkv_proj.weight": "model-00001-of-00004.safetensors",
|
| 200 |
+
"model.norm.weight": "model-00003-of-00004.safetensors"
|
| 201 |
+
}
|
| 202 |
+
}
|
optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b901262891c97b18b30ecc1125283e1f76958450e507ab529889c61a9101ae0
|
| 3 |
+
size 30566201634
|
rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9196a1e708bf24d6abba41cce3f8558820acc3e50f9394c5955e29eb41ffea3d
|
| 3 |
+
size 14244
|
scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d443fe92d884b41152079d0a798667000e213b61ace3999525c785c947b9fd4f
|
| 3 |
+
size 1064
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"eos_token": {
|
| 10 |
+
"content": "<|endoftext|>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"pad_token": "<|endoftext|>",
|
| 17 |
+
"unk_token": {
|
| 18 |
+
"content": "<unk>",
|
| 19 |
+
"lstrip": false,
|
| 20 |
+
"normalized": false,
|
| 21 |
+
"rstrip": false,
|
| 22 |
+
"single_word": false
|
| 23 |
+
}
|
| 24 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_bos_token": true,
|
| 3 |
+
"add_eos_token": false,
|
| 4 |
+
"added_tokens_decoder": {
|
| 5 |
+
"0": {
|
| 6 |
+
"content": "<unk>",
|
| 7 |
+
"lstrip": false,
|
| 8 |
+
"normalized": false,
|
| 9 |
+
"rstrip": false,
|
| 10 |
+
"single_word": false,
|
| 11 |
+
"special": true
|
| 12 |
+
},
|
| 13 |
+
"1": {
|
| 14 |
+
"content": "<s>",
|
| 15 |
+
"lstrip": false,
|
| 16 |
+
"normalized": false,
|
| 17 |
+
"rstrip": false,
|
| 18 |
+
"single_word": false,
|
| 19 |
+
"special": true
|
| 20 |
+
},
|
| 21 |
+
"2": {
|
| 22 |
+
"content": "</s>",
|
| 23 |
+
"lstrip": false,
|
| 24 |
+
"normalized": false,
|
| 25 |
+
"rstrip": true,
|
| 26 |
+
"single_word": false,
|
| 27 |
+
"special": false
|
| 28 |
+
},
|
| 29 |
+
"32000": {
|
| 30 |
+
"content": "<|endoftext|>",
|
| 31 |
+
"lstrip": false,
|
| 32 |
+
"normalized": false,
|
| 33 |
+
"rstrip": false,
|
| 34 |
+
"single_word": false,
|
| 35 |
+
"special": true
|
| 36 |
+
},
|
| 37 |
+
"32001": {
|
| 38 |
+
"content": "<|assistant|>",
|
| 39 |
+
"lstrip": false,
|
| 40 |
+
"normalized": false,
|
| 41 |
+
"rstrip": true,
|
| 42 |
+
"single_word": false,
|
| 43 |
+
"special": true
|
| 44 |
+
},
|
| 45 |
+
"32002": {
|
| 46 |
+
"content": "<|placeholder1|>",
|
| 47 |
+
"lstrip": false,
|
| 48 |
+
"normalized": false,
|
| 49 |
+
"rstrip": true,
|
| 50 |
+
"single_word": false,
|
| 51 |
+
"special": true
|
| 52 |
+
},
|
| 53 |
+
"32003": {
|
| 54 |
+
"content": "<|placeholder2|>",
|
| 55 |
+
"lstrip": false,
|
| 56 |
+
"normalized": false,
|
| 57 |
+
"rstrip": true,
|
| 58 |
+
"single_word": false,
|
| 59 |
+
"special": true
|
| 60 |
+
},
|
| 61 |
+
"32004": {
|
| 62 |
+
"content": "<|placeholder3|>",
|
| 63 |
+
"lstrip": false,
|
| 64 |
+
"normalized": false,
|
| 65 |
+
"rstrip": true,
|
| 66 |
+
"single_word": false,
|
| 67 |
+
"special": true
|
| 68 |
+
},
|
| 69 |
+
"32005": {
|
| 70 |
+
"content": "<|placeholder4|>",
|
| 71 |
+
"lstrip": false,
|
| 72 |
+
"normalized": false,
|
| 73 |
+
"rstrip": true,
|
| 74 |
+
"single_word": false,
|
| 75 |
+
"special": true
|
| 76 |
+
},
|
| 77 |
+
"32006": {
|
| 78 |
+
"content": "<|system|>",
|
| 79 |
+
"lstrip": false,
|
| 80 |
+
"normalized": false,
|
| 81 |
+
"rstrip": true,
|
| 82 |
+
"single_word": false,
|
| 83 |
+
"special": true
|
| 84 |
+
},
|
| 85 |
+
"32007": {
|
| 86 |
+
"content": "<|end|>",
|
| 87 |
+
"lstrip": false,
|
| 88 |
+
"normalized": false,
|
| 89 |
+
"rstrip": true,
|
| 90 |
+
"single_word": false,
|
| 91 |
+
"special": true
|
| 92 |
+
},
|
| 93 |
+
"32008": {
|
| 94 |
+
"content": "<|placeholder5|>",
|
| 95 |
+
"lstrip": false,
|
| 96 |
+
"normalized": false,
|
| 97 |
+
"rstrip": true,
|
| 98 |
+
"single_word": false,
|
| 99 |
+
"special": true
|
| 100 |
+
},
|
| 101 |
+
"32009": {
|
| 102 |
+
"content": "<|placeholder6|>",
|
| 103 |
+
"lstrip": false,
|
| 104 |
+
"normalized": false,
|
| 105 |
+
"rstrip": true,
|
| 106 |
+
"single_word": false,
|
| 107 |
+
"special": true
|
| 108 |
+
},
|
| 109 |
+
"32010": {
|
| 110 |
+
"content": "<|user|>",
|
| 111 |
+
"lstrip": false,
|
| 112 |
+
"normalized": false,
|
| 113 |
+
"rstrip": true,
|
| 114 |
+
"single_word": false,
|
| 115 |
+
"special": true
|
| 116 |
+
}
|
| 117 |
+
},
|
| 118 |
+
"bos_token": "<s>",
|
| 119 |
+
"chat_template": "{{ bos_token }}{% for message in messages %}{% if (message['role'] == 'user') %}{{'<|user|>' + '\n' + message['content'] + '<|end|>' + '\n' + '<|assistant|>' + '\n'}}{% elif (message['role'] == 'assistant') %}{{message['content'] + '<|end|>' + '\n'}}{% endif %}{% endfor %}",
|
| 120 |
+
"clean_up_tokenization_spaces": false,
|
| 121 |
+
"eos_token": "<|endoftext|>",
|
| 122 |
+
"legacy": false,
|
| 123 |
+
"model_max_length": 4096,
|
| 124 |
+
"pad_token": "<|endoftext|>",
|
| 125 |
+
"padding_side": "left",
|
| 126 |
+
"sp_model_kwargs": {},
|
| 127 |
+
"tokenizer_class": "LlamaTokenizer",
|
| 128 |
+
"unk_token": "<unk>",
|
| 129 |
+
"use_default_system_prompt": false
|
| 130 |
+
}
|
trainer_state.json
ADDED
|
@@ -0,0 +1,543 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_metric": null,
|
| 3 |
+
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 2.957146863129987,
|
| 5 |
+
"eval_steps": 500,
|
| 6 |
+
"global_step": 14500,
|
| 7 |
+
"is_hyper_param_search": false,
|
| 8 |
+
"is_local_process_zero": true,
|
| 9 |
+
"is_world_process_zero": true,
|
| 10 |
+
"log_history": [
|
| 11 |
+
{
|
| 12 |
+
"epoch": 0.10197058148724093,
|
| 13 |
+
"grad_norm": 3.7134974002838135,
|
| 14 |
+
"learning_rate": 4.8313957441022504e-05,
|
| 15 |
+
"log_odds_chosen": 0.2222207486629486,
|
| 16 |
+
"log_odds_ratio": -0.6578092575073242,
|
| 17 |
+
"logits/chosen": -0.44847217202186584,
|
| 18 |
+
"logits/rejected": -0.7988718152046204,
|
| 19 |
+
"logps/chosen": -1.3054556846618652,
|
| 20 |
+
"logps/rejected": -1.482597827911377,
|
| 21 |
+
"loss": 1.2524,
|
| 22 |
+
"nll_loss": 1.1866657733917236,
|
| 23 |
+
"rewards/accuracies": 0.5920000076293945,
|
| 24 |
+
"rewards/chosen": -0.13054557144641876,
|
| 25 |
+
"rewards/margins": 0.01771421544253826,
|
| 26 |
+
"rewards/rejected": -0.14825978875160217,
|
| 27 |
+
"step": 500
|
| 28 |
+
},
|
| 29 |
+
{
|
| 30 |
+
"epoch": 0.20394116297448187,
|
| 31 |
+
"grad_norm": 2.004502058029175,
|
| 32 |
+
"learning_rate": 4.6614317764633906e-05,
|
| 33 |
+
"log_odds_chosen": 0.35723409056663513,
|
| 34 |
+
"log_odds_ratio": -0.629149317741394,
|
| 35 |
+
"logits/chosen": -1.3414314985275269,
|
| 36 |
+
"logits/rejected": -1.5448620319366455,
|
| 37 |
+
"logps/chosen": -1.255893588066101,
|
| 38 |
+
"logps/rejected": -1.5381147861480713,
|
| 39 |
+
"loss": 1.0135,
|
| 40 |
+
"nll_loss": 0.9506049752235413,
|
| 41 |
+
"rewards/accuracies": 0.609000027179718,
|
| 42 |
+
"rewards/chosen": -0.12558937072753906,
|
| 43 |
+
"rewards/margins": 0.028222110122442245,
|
| 44 |
+
"rewards/rejected": -0.1538114696741104,
|
| 45 |
+
"step": 1000
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.3059117444617228,
|
| 49 |
+
"grad_norm": 1.5967992544174194,
|
| 50 |
+
"learning_rate": 4.4914678088245294e-05,
|
| 51 |
+
"log_odds_chosen": 0.45203697681427,
|
| 52 |
+
"log_odds_ratio": -0.6168015003204346,
|
| 53 |
+
"logits/chosen": -1.578227162361145,
|
| 54 |
+
"logits/rejected": -1.7268493175506592,
|
| 55 |
+
"logps/chosen": -1.1664234399795532,
|
| 56 |
+
"logps/rejected": -1.515409231185913,
|
| 57 |
+
"loss": 0.9167,
|
| 58 |
+
"nll_loss": 0.8550541400909424,
|
| 59 |
+
"rewards/accuracies": 0.6147500276565552,
|
| 60 |
+
"rewards/chosen": -0.11664234101772308,
|
| 61 |
+
"rewards/margins": 0.034898579120635986,
|
| 62 |
+
"rewards/rejected": -0.15154090523719788,
|
| 63 |
+
"step": 1500
|
| 64 |
+
},
|
| 65 |
+
{
|
| 66 |
+
"epoch": 0.40788232594896373,
|
| 67 |
+
"grad_norm": 1.4804637432098389,
|
| 68 |
+
"learning_rate": 4.321503841185669e-05,
|
| 69 |
+
"log_odds_chosen": 0.5433089733123779,
|
| 70 |
+
"log_odds_ratio": -0.605213463306427,
|
| 71 |
+
"logits/chosen": -1.7934901714324951,
|
| 72 |
+
"logits/rejected": -1.9556081295013428,
|
| 73 |
+
"logps/chosen": -1.108548879623413,
|
| 74 |
+
"logps/rejected": -1.5245370864868164,
|
| 75 |
+
"loss": 0.8516,
|
| 76 |
+
"nll_loss": 0.7910776734352112,
|
| 77 |
+
"rewards/accuracies": 0.628250002861023,
|
| 78 |
+
"rewards/chosen": -0.11085488647222519,
|
| 79 |
+
"rewards/margins": 0.04159883037209511,
|
| 80 |
+
"rewards/rejected": -0.1524537056684494,
|
| 81 |
+
"step": 2000
|
| 82 |
+
},
|
| 83 |
+
{
|
| 84 |
+
"epoch": 0.5098529074362047,
|
| 85 |
+
"grad_norm": 1.2357059717178345,
|
| 86 |
+
"learning_rate": 4.151539873546808e-05,
|
| 87 |
+
"log_odds_chosen": 0.6769158244132996,
|
| 88 |
+
"log_odds_ratio": -0.5792952179908752,
|
| 89 |
+
"logits/chosen": -1.9547886848449707,
|
| 90 |
+
"logits/rejected": -2.1045212745666504,
|
| 91 |
+
"logps/chosen": -1.04926335811615,
|
| 92 |
+
"logps/rejected": -1.554589033126831,
|
| 93 |
+
"loss": 0.7987,
|
| 94 |
+
"nll_loss": 0.7408133745193481,
|
| 95 |
+
"rewards/accuracies": 0.656499981880188,
|
| 96 |
+
"rewards/chosen": -0.10492634028196335,
|
| 97 |
+
"rewards/margins": 0.050532568246126175,
|
| 98 |
+
"rewards/rejected": -0.15545891225337982,
|
| 99 |
+
"step": 2500
|
| 100 |
+
},
|
| 101 |
+
{
|
| 102 |
+
"epoch": 0.6118234889234456,
|
| 103 |
+
"grad_norm": 1.753408432006836,
|
| 104 |
+
"learning_rate": 3.981575905907948e-05,
|
| 105 |
+
"log_odds_chosen": 0.7549012303352356,
|
| 106 |
+
"log_odds_ratio": -0.5683782696723938,
|
| 107 |
+
"logits/chosen": -2.111973285675049,
|
| 108 |
+
"logits/rejected": -2.268782615661621,
|
| 109 |
+
"logps/chosen": -1.0075668096542358,
|
| 110 |
+
"logps/rejected": -1.5643786191940308,
|
| 111 |
+
"loss": 0.7492,
|
| 112 |
+
"nll_loss": 0.6923843026161194,
|
| 113 |
+
"rewards/accuracies": 0.640999972820282,
|
| 114 |
+
"rewards/chosen": -0.1007566824555397,
|
| 115 |
+
"rewards/margins": 0.055681198835372925,
|
| 116 |
+
"rewards/rejected": -0.15643785893917084,
|
| 117 |
+
"step": 3000
|
| 118 |
+
},
|
| 119 |
+
{
|
| 120 |
+
"epoch": 0.7137940704106865,
|
| 121 |
+
"grad_norm": 1.8108484745025635,
|
| 122 |
+
"learning_rate": 3.811611938269087e-05,
|
| 123 |
+
"log_odds_chosen": 0.7971724271774292,
|
| 124 |
+
"log_odds_ratio": -0.5602663159370422,
|
| 125 |
+
"logits/chosen": -2.263519525527954,
|
| 126 |
+
"logits/rejected": -2.3989477157592773,
|
| 127 |
+
"logps/chosen": -0.9775514006614685,
|
| 128 |
+
"logps/rejected": -1.5509178638458252,
|
| 129 |
+
"loss": 0.7398,
|
| 130 |
+
"nll_loss": 0.6837689280509949,
|
| 131 |
+
"rewards/accuracies": 0.6542500257492065,
|
| 132 |
+
"rewards/chosen": -0.09775513410568237,
|
| 133 |
+
"rewards/margins": 0.05733664333820343,
|
| 134 |
+
"rewards/rejected": -0.1550917774438858,
|
| 135 |
+
"step": 3500
|
| 136 |
+
},
|
| 137 |
+
{
|
| 138 |
+
"epoch": 0.8157646518979275,
|
| 139 |
+
"grad_norm": 0.9750083684921265,
|
| 140 |
+
"learning_rate": 3.641647970630227e-05,
|
| 141 |
+
"log_odds_chosen": 0.8391203880310059,
|
| 142 |
+
"log_odds_ratio": -0.5608585476875305,
|
| 143 |
+
"logits/chosen": -2.2849109172821045,
|
| 144 |
+
"logits/rejected": -2.413522720336914,
|
| 145 |
+
"logps/chosen": -0.9366417527198792,
|
| 146 |
+
"logps/rejected": -1.5379432439804077,
|
| 147 |
+
"loss": 0.708,
|
| 148 |
+
"nll_loss": 0.6518718600273132,
|
| 149 |
+
"rewards/accuracies": 0.6554999947547913,
|
| 150 |
+
"rewards/chosen": -0.09366416931152344,
|
| 151 |
+
"rewards/margins": 0.06013017147779465,
|
| 152 |
+
"rewards/rejected": -0.1537943333387375,
|
| 153 |
+
"step": 4000
|
| 154 |
+
},
|
| 155 |
+
{
|
| 156 |
+
"epoch": 0.9177352333851684,
|
| 157 |
+
"grad_norm": 1.1859960556030273,
|
| 158 |
+
"learning_rate": 3.471684002991366e-05,
|
| 159 |
+
"log_odds_chosen": 0.9322497844696045,
|
| 160 |
+
"log_odds_ratio": -0.5479665398597717,
|
| 161 |
+
"logits/chosen": -2.4118337631225586,
|
| 162 |
+
"logits/rejected": -2.5123636722564697,
|
| 163 |
+
"logps/chosen": -0.9224444031715393,
|
| 164 |
+
"logps/rejected": -1.572380781173706,
|
| 165 |
+
"loss": 0.6784,
|
| 166 |
+
"nll_loss": 0.6236190795898438,
|
| 167 |
+
"rewards/accuracies": 0.6604999899864197,
|
| 168 |
+
"rewards/chosen": -0.09224444627761841,
|
| 169 |
+
"rewards/margins": 0.06499364972114563,
|
| 170 |
+
"rewards/rejected": -0.15723808109760284,
|
| 171 |
+
"step": 4500
|
| 172 |
+
},
|
| 173 |
+
{
|
| 174 |
+
"epoch": 1.0197058148724094,
|
| 175 |
+
"grad_norm": 1.545590877532959,
|
| 176 |
+
"learning_rate": 3.301720035352505e-05,
|
| 177 |
+
"log_odds_chosen": 1.231202244758606,
|
| 178 |
+
"log_odds_ratio": -0.4899790585041046,
|
| 179 |
+
"logits/chosen": -2.496483564376831,
|
| 180 |
+
"logits/rejected": -2.5658133029937744,
|
| 181 |
+
"logps/chosen": -0.8024629950523376,
|
| 182 |
+
"logps/rejected": -1.6277039051055908,
|
| 183 |
+
"loss": 0.5995,
|
| 184 |
+
"nll_loss": 0.5504766702651978,
|
| 185 |
+
"rewards/accuracies": 0.7229999899864197,
|
| 186 |
+
"rewards/chosen": -0.08024629950523376,
|
| 187 |
+
"rewards/margins": 0.08252407610416412,
|
| 188 |
+
"rewards/rejected": -0.16277039051055908,
|
| 189 |
+
"step": 5000
|
| 190 |
+
},
|
| 191 |
+
{
|
| 192 |
+
"epoch": 1.1216763963596503,
|
| 193 |
+
"grad_norm": 1.1230560541152954,
|
| 194 |
+
"learning_rate": 3.132095995648923e-05,
|
| 195 |
+
"log_odds_chosen": 2.3458921909332275,
|
| 196 |
+
"log_odds_ratio": -0.2528613209724426,
|
| 197 |
+
"logits/chosen": -2.7195615768432617,
|
| 198 |
+
"logits/rejected": -2.732959270477295,
|
| 199 |
+
"logps/chosen": -0.4438183605670929,
|
| 200 |
+
"logps/rejected": -1.9251257181167603,
|
| 201 |
+
"loss": 0.3371,
|
| 202 |
+
"nll_loss": 0.311859130859375,
|
| 203 |
+
"rewards/accuracies": 0.9585000276565552,
|
| 204 |
+
"rewards/chosen": -0.04438183456659317,
|
| 205 |
+
"rewards/margins": 0.14813074469566345,
|
| 206 |
+
"rewards/rejected": -0.19251257181167603,
|
| 207 |
+
"step": 5500
|
| 208 |
+
},
|
| 209 |
+
{
|
| 210 |
+
"epoch": 1.2236469778468912,
|
| 211 |
+
"grad_norm": 1.688524603843689,
|
| 212 |
+
"learning_rate": 2.9621320280100618e-05,
|
| 213 |
+
"log_odds_chosen": 2.3827967643737793,
|
| 214 |
+
"log_odds_ratio": -0.25957345962524414,
|
| 215 |
+
"logits/chosen": -2.720367193222046,
|
| 216 |
+
"logits/rejected": -2.767230987548828,
|
| 217 |
+
"logps/chosen": -0.44125357270240784,
|
| 218 |
+
"logps/rejected": -1.953121304512024,
|
| 219 |
+
"loss": 0.334,
|
| 220 |
+
"nll_loss": 0.3080129623413086,
|
| 221 |
+
"rewards/accuracies": 0.9617499709129333,
|
| 222 |
+
"rewards/chosen": -0.04412535950541496,
|
| 223 |
+
"rewards/margins": 0.1511867791414261,
|
| 224 |
+
"rewards/rejected": -0.19531212747097015,
|
| 225 |
+
"step": 6000
|
| 226 |
+
},
|
| 227 |
+
{
|
| 228 |
+
"epoch": 1.3256175593341322,
|
| 229 |
+
"grad_norm": 1.2772260904312134,
|
| 230 |
+
"learning_rate": 2.792507988306479e-05,
|
| 231 |
+
"log_odds_chosen": 2.2994472980499268,
|
| 232 |
+
"log_odds_ratio": -0.26726964116096497,
|
| 233 |
+
"logits/chosen": -2.7083919048309326,
|
| 234 |
+
"logits/rejected": -2.772411823272705,
|
| 235 |
+
"logps/chosen": -0.45140743255615234,
|
| 236 |
+
"logps/rejected": -1.911563754081726,
|
| 237 |
+
"loss": 0.3471,
|
| 238 |
+
"nll_loss": 0.32037463784217834,
|
| 239 |
+
"rewards/accuracies": 0.9572499990463257,
|
| 240 |
+
"rewards/chosen": -0.045140739530324936,
|
| 241 |
+
"rewards/margins": 0.14601562917232513,
|
| 242 |
+
"rewards/rejected": -0.19115637242794037,
|
| 243 |
+
"step": 6500
|
| 244 |
+
},
|
| 245 |
+
{
|
| 246 |
+
"epoch": 1.427588140821373,
|
| 247 |
+
"grad_norm": 1.3095237016677856,
|
| 248 |
+
"learning_rate": 2.622544020667619e-05,
|
| 249 |
+
"log_odds_chosen": 2.344271659851074,
|
| 250 |
+
"log_odds_ratio": -0.27038824558258057,
|
| 251 |
+
"logits/chosen": -2.74202823638916,
|
| 252 |
+
"logits/rejected": -2.7950618267059326,
|
| 253 |
+
"logps/chosen": -0.42455798387527466,
|
| 254 |
+
"logps/rejected": -1.880998134613037,
|
| 255 |
+
"loss": 0.3258,
|
| 256 |
+
"nll_loss": 0.29875221848487854,
|
| 257 |
+
"rewards/accuracies": 0.952750027179718,
|
| 258 |
+
"rewards/chosen": -0.042455799877643585,
|
| 259 |
+
"rewards/margins": 0.14564399421215057,
|
| 260 |
+
"rewards/rejected": -0.18809978663921356,
|
| 261 |
+
"step": 7000
|
| 262 |
+
},
|
| 263 |
+
{
|
| 264 |
+
"epoch": 1.529558722308614,
|
| 265 |
+
"grad_norm": 1.1868637800216675,
|
| 266 |
+
"learning_rate": 2.452580053028758e-05,
|
| 267 |
+
"log_odds_chosen": 2.3831584453582764,
|
| 268 |
+
"log_odds_ratio": -0.26601019501686096,
|
| 269 |
+
"logits/chosen": -2.7699034214019775,
|
| 270 |
+
"logits/rejected": -2.8358941078186035,
|
| 271 |
+
"logps/chosen": -0.43710044026374817,
|
| 272 |
+
"logps/rejected": -1.9111379384994507,
|
| 273 |
+
"loss": 0.3313,
|
| 274 |
+
"nll_loss": 0.3046627938747406,
|
| 275 |
+
"rewards/accuracies": 0.9614999890327454,
|
| 276 |
+
"rewards/chosen": -0.043710045516490936,
|
| 277 |
+
"rewards/margins": 0.147403746843338,
|
| 278 |
+
"rewards/rejected": -0.19111379981040955,
|
| 279 |
+
"step": 7500
|
| 280 |
+
},
|
| 281 |
+
{
|
| 282 |
+
"epoch": 1.631529303795855,
|
| 283 |
+
"grad_norm": 1.8944728374481201,
|
| 284 |
+
"learning_rate": 2.2826160853898975e-05,
|
| 285 |
+
"log_odds_chosen": 2.470182180404663,
|
| 286 |
+
"log_odds_ratio": -0.2623186409473419,
|
| 287 |
+
"logits/chosen": -2.7561872005462646,
|
| 288 |
+
"logits/rejected": -2.7932546138763428,
|
| 289 |
+
"logps/chosen": -0.43067559599876404,
|
| 290 |
+
"logps/rejected": -1.9674099683761597,
|
| 291 |
+
"loss": 0.325,
|
| 292 |
+
"nll_loss": 0.2987590432167053,
|
| 293 |
+
"rewards/accuracies": 0.956250011920929,
|
| 294 |
+
"rewards/chosen": -0.043067559599876404,
|
| 295 |
+
"rewards/margins": 0.1536734402179718,
|
| 296 |
+
"rewards/rejected": -0.196740984916687,
|
| 297 |
+
"step": 8000
|
| 298 |
+
},
|
| 299 |
+
{
|
| 300 |
+
"epoch": 1.7334998852830958,
|
| 301 |
+
"grad_norm": 1.0028513669967651,
|
| 302 |
+
"learning_rate": 2.112652117751037e-05,
|
| 303 |
+
"log_odds_chosen": 2.5264878273010254,
|
| 304 |
+
"log_odds_ratio": -0.25732529163360596,
|
| 305 |
+
"logits/chosen": -2.8135437965393066,
|
| 306 |
+
"logits/rejected": -2.852220058441162,
|
| 307 |
+
"logps/chosen": -0.41469958424568176,
|
| 308 |
+
"logps/rejected": -1.949130654335022,
|
| 309 |
+
"loss": 0.3154,
|
| 310 |
+
"nll_loss": 0.2897038757801056,
|
| 311 |
+
"rewards/accuracies": 0.9607499837875366,
|
| 312 |
+
"rewards/chosen": -0.04146995395421982,
|
| 313 |
+
"rewards/margins": 0.1534431129693985,
|
| 314 |
+
"rewards/rejected": -0.1949130743741989,
|
| 315 |
+
"step": 8500
|
| 316 |
+
},
|
| 317 |
+
{
|
| 318 |
+
"epoch": 1.8354704667703368,
|
| 319 |
+
"grad_norm": 0.9701994061470032,
|
| 320 |
+
"learning_rate": 1.9426881501121764e-05,
|
| 321 |
+
"log_odds_chosen": 2.3967556953430176,
|
| 322 |
+
"log_odds_ratio": -0.27115362882614136,
|
| 323 |
+
"logits/chosen": -2.7224009037017822,
|
| 324 |
+
"logits/rejected": -2.7754287719726562,
|
| 325 |
+
"logps/chosen": -0.4164165258407593,
|
| 326 |
+
"logps/rejected": -1.8547587394714355,
|
| 327 |
+
"loss": 0.3216,
|
| 328 |
+
"nll_loss": 0.29452264308929443,
|
| 329 |
+
"rewards/accuracies": 0.9505000114440918,
|
| 330 |
+
"rewards/chosen": -0.041641656309366226,
|
| 331 |
+
"rewards/margins": 0.1438342034816742,
|
| 332 |
+
"rewards/rejected": -0.18547587096691132,
|
| 333 |
+
"step": 9000
|
| 334 |
+
},
|
| 335 |
+
{
|
| 336 |
+
"epoch": 1.9374410482575777,
|
| 337 |
+
"grad_norm": 0.7663633823394775,
|
| 338 |
+
"learning_rate": 1.7727241824733156e-05,
|
| 339 |
+
"log_odds_chosen": 2.5609018802642822,
|
| 340 |
+
"log_odds_ratio": -0.26221731305122375,
|
| 341 |
+
"logits/chosen": -2.8092148303985596,
|
| 342 |
+
"logits/rejected": -2.836301803588867,
|
| 343 |
+
"logps/chosen": -0.3934709429740906,
|
| 344 |
+
"logps/rejected": -1.921687364578247,
|
| 345 |
+
"loss": 0.3055,
|
| 346 |
+
"nll_loss": 0.27924445271492004,
|
| 347 |
+
"rewards/accuracies": 0.9557499885559082,
|
| 348 |
+
"rewards/chosen": -0.0393470898270607,
|
| 349 |
+
"rewards/margins": 0.1528216451406479,
|
| 350 |
+
"rewards/rejected": -0.192168727517128,
|
| 351 |
+
"step": 9500
|
| 352 |
+
},
|
| 353 |
+
{
|
| 354 |
+
"epoch": 2.039411629744819,
|
| 355 |
+
"grad_norm": 0.6147420406341553,
|
| 356 |
+
"learning_rate": 1.602760214834455e-05,
|
| 357 |
+
"log_odds_chosen": 3.2184579372406006,
|
| 358 |
+
"log_odds_ratio": -0.18695417046546936,
|
| 359 |
+
"logits/chosen": -3.0869991779327393,
|
| 360 |
+
"logits/rejected": -3.0871739387512207,
|
| 361 |
+
"logps/chosen": -0.3052729070186615,
|
| 362 |
+
"logps/rejected": -2.1187493801116943,
|
| 363 |
+
"loss": 0.2355,
|
| 364 |
+
"nll_loss": 0.2168271690607071,
|
| 365 |
+
"rewards/accuracies": 0.9782500267028809,
|
| 366 |
+
"rewards/chosen": -0.03052728995680809,
|
| 367 |
+
"rewards/margins": 0.18134763836860657,
|
| 368 |
+
"rewards/rejected": -0.21187493205070496,
|
| 369 |
+
"step": 10000
|
| 370 |
+
},
|
| 371 |
+
{
|
| 372 |
+
"epoch": 2.1413822112320595,
|
| 373 |
+
"grad_norm": 0.8968594074249268,
|
| 374 |
+
"learning_rate": 1.4327962471955947e-05,
|
| 375 |
+
"log_odds_chosen": 4.477025032043457,
|
| 376 |
+
"log_odds_ratio": -0.07086528092622757,
|
| 377 |
+
"logits/chosen": -3.4468417167663574,
|
| 378 |
+
"logits/rejected": -3.4328184127807617,
|
| 379 |
+
"logps/chosen": -0.1490449756383896,
|
| 380 |
+
"logps/rejected": -2.5894744396209717,
|
| 381 |
+
"loss": 0.1219,
|
| 382 |
+
"nll_loss": 0.11482993513345718,
|
| 383 |
+
"rewards/accuracies": 0.9940000176429749,
|
| 384 |
+
"rewards/chosen": -0.014904499053955078,
|
| 385 |
+
"rewards/margins": 0.24404291808605194,
|
| 386 |
+
"rewards/rejected": -0.2589474618434906,
|
| 387 |
+
"step": 10500
|
| 388 |
+
},
|
| 389 |
+
{
|
| 390 |
+
"epoch": 2.2433527927193007,
|
| 391 |
+
"grad_norm": 0.7543737292289734,
|
| 392 |
+
"learning_rate": 1.262832279556734e-05,
|
| 393 |
+
"log_odds_chosen": 4.517333030700684,
|
| 394 |
+
"log_odds_ratio": -0.07165472209453583,
|
| 395 |
+
"logits/chosen": -3.4554872512817383,
|
| 396 |
+
"logits/rejected": -3.4509389400482178,
|
| 397 |
+
"logps/chosen": -0.151213601231575,
|
| 398 |
+
"logps/rejected": -2.6054680347442627,
|
| 399 |
+
"loss": 0.1234,
|
| 400 |
+
"nll_loss": 0.11624418199062347,
|
| 401 |
+
"rewards/accuracies": 0.9909999966621399,
|
| 402 |
+
"rewards/chosen": -0.015121362172067165,
|
| 403 |
+
"rewards/margins": 0.24542541801929474,
|
| 404 |
+
"rewards/rejected": -0.2605467736721039,
|
| 405 |
+
"step": 11000
|
| 406 |
+
},
|
| 407 |
+
{
|
| 408 |
+
"epoch": 2.3453233742065414,
|
| 409 |
+
"grad_norm": 0.7870352864265442,
|
| 410 |
+
"learning_rate": 1.0928683119178734e-05,
|
| 411 |
+
"log_odds_chosen": 4.504143238067627,
|
| 412 |
+
"log_odds_ratio": -0.0680876150727272,
|
| 413 |
+
"logits/chosen": -3.3482730388641357,
|
| 414 |
+
"logits/rejected": -3.362037181854248,
|
| 415 |
+
"logps/chosen": -0.15102657675743103,
|
| 416 |
+
"logps/rejected": -2.60615873336792,
|
| 417 |
+
"loss": 0.1227,
|
| 418 |
+
"nll_loss": 0.11586029082536697,
|
| 419 |
+
"rewards/accuracies": 0.9937499761581421,
|
| 420 |
+
"rewards/chosen": -0.015102657489478588,
|
| 421 |
+
"rewards/margins": 0.24551323056221008,
|
| 422 |
+
"rewards/rejected": -0.26061588525772095,
|
| 423 |
+
"step": 11500
|
| 424 |
+
},
|
| 425 |
+
{
|
| 426 |
+
"epoch": 2.4472939556937825,
|
| 427 |
+
"grad_norm": 0.621370792388916,
|
| 428 |
+
"learning_rate": 9.22904344279013e-06,
|
| 429 |
+
"log_odds_chosen": 4.464700698852539,
|
| 430 |
+
"log_odds_ratio": -0.07176685333251953,
|
| 431 |
+
"logits/chosen": -3.3470752239227295,
|
| 432 |
+
"logits/rejected": -3.3573222160339355,
|
| 433 |
+
"logps/chosen": -0.14544738829135895,
|
| 434 |
+
"logps/rejected": -2.5615859031677246,
|
| 435 |
+
"loss": 0.12,
|
| 436 |
+
"nll_loss": 0.11279115825891495,
|
| 437 |
+
"rewards/accuracies": 0.9892500042915344,
|
| 438 |
+
"rewards/chosen": -0.014544738456606865,
|
| 439 |
+
"rewards/margins": 0.24161386489868164,
|
| 440 |
+
"rewards/rejected": -0.25615859031677246,
|
| 441 |
+
"step": 12000
|
| 442 |
+
},
|
| 443 |
+
{
|
| 444 |
+
"epoch": 2.549264537181023,
|
| 445 |
+
"grad_norm": 0.8108669519424438,
|
| 446 |
+
"learning_rate": 7.529403766401523e-06,
|
| 447 |
+
"log_odds_chosen": 4.566464424133301,
|
| 448 |
+
"log_odds_ratio": -0.06760641187429428,
|
| 449 |
+
"logits/chosen": -3.3751792907714844,
|
| 450 |
+
"logits/rejected": -3.371516227722168,
|
| 451 |
+
"logps/chosen": -0.14034530520439148,
|
| 452 |
+
"logps/rejected": -2.578918933868408,
|
| 453 |
+
"loss": 0.1152,
|
| 454 |
+
"nll_loss": 0.1084122508764267,
|
| 455 |
+
"rewards/accuracies": 0.9932500123977661,
|
| 456 |
+
"rewards/chosen": -0.014034531079232693,
|
| 457 |
+
"rewards/margins": 0.24385739862918854,
|
| 458 |
+
"rewards/rejected": -0.2578918933868408,
|
| 459 |
+
"step": 12500
|
| 460 |
+
},
|
| 461 |
+
{
|
| 462 |
+
"epoch": 2.6512351186682643,
|
| 463 |
+
"grad_norm": 1.0674421787261963,
|
| 464 |
+
"learning_rate": 5.829764090012918e-06,
|
| 465 |
+
"log_odds_chosen": 4.609077453613281,
|
| 466 |
+
"log_odds_ratio": -0.06549856811761856,
|
| 467 |
+
"logits/chosen": -3.320563316345215,
|
| 468 |
+
"logits/rejected": -3.3299055099487305,
|
| 469 |
+
"logps/chosen": -0.1383795589208603,
|
| 470 |
+
"logps/rejected": -2.625169515609741,
|
| 471 |
+
"loss": 0.1144,
|
| 472 |
+
"nll_loss": 0.10785890370607376,
|
| 473 |
+
"rewards/accuracies": 0.9932500123977661,
|
| 474 |
+
"rewards/chosen": -0.013837955892086029,
|
| 475 |
+
"rewards/margins": 0.2486789971590042,
|
| 476 |
+
"rewards/rejected": -0.26251694560050964,
|
| 477 |
+
"step": 13000
|
| 478 |
+
},
|
| 479 |
+
{
|
| 480 |
+
"epoch": 2.753205700155505,
|
| 481 |
+
"grad_norm": 0.9907452464103699,
|
| 482 |
+
"learning_rate": 4.1301244136243116e-06,
|
| 483 |
+
"log_odds_chosen": 4.6315083503723145,
|
| 484 |
+
"log_odds_ratio": -0.06764637678861618,
|
| 485 |
+
"logits/chosen": -3.35665225982666,
|
| 486 |
+
"logits/rejected": -3.365429639816284,
|
| 487 |
+
"logps/chosen": -0.14084672927856445,
|
| 488 |
+
"logps/rejected": -2.6286959648132324,
|
| 489 |
+
"loss": 0.1155,
|
| 490 |
+
"nll_loss": 0.10870947688817978,
|
| 491 |
+
"rewards/accuracies": 0.9917500019073486,
|
| 492 |
+
"rewards/chosen": -0.01408467348664999,
|
| 493 |
+
"rewards/margins": 0.24878491461277008,
|
| 494 |
+
"rewards/rejected": -0.26286956667900085,
|
| 495 |
+
"step": 13500
|
| 496 |
+
},
|
| 497 |
+
{
|
| 498 |
+
"epoch": 2.855176281642746,
|
| 499 |
+
"grad_norm": 1.0650596618652344,
|
| 500 |
+
"learning_rate": 2.4304847372357063e-06,
|
| 501 |
+
"log_odds_chosen": 4.683770656585693,
|
| 502 |
+
"log_odds_ratio": -0.06298153102397919,
|
| 503 |
+
"logits/chosen": -3.343515634536743,
|
| 504 |
+
"logits/rejected": -3.349973201751709,
|
| 505 |
+
"logps/chosen": -0.1364256739616394,
|
| 506 |
+
"logps/rejected": -2.6349427700042725,
|
| 507 |
+
"loss": 0.111,
|
| 508 |
+
"nll_loss": 0.10465656220912933,
|
| 509 |
+
"rewards/accuracies": 0.9919999837875366,
|
| 510 |
+
"rewards/chosen": -0.013642566278576851,
|
| 511 |
+
"rewards/margins": 0.24985171854496002,
|
| 512 |
+
"rewards/rejected": -0.26349425315856934,
|
| 513 |
+
"step": 14000
|
| 514 |
+
},
|
| 515 |
+
{
|
| 516 |
+
"epoch": 2.957146863129987,
|
| 517 |
+
"grad_norm": 0.7051740288734436,
|
| 518 |
+
"learning_rate": 7.308450608471004e-07,
|
| 519 |
+
"log_odds_chosen": 4.733154296875,
|
| 520 |
+
"log_odds_ratio": -0.06264213472604752,
|
| 521 |
+
"logits/chosen": -3.384558916091919,
|
| 522 |
+
"logits/rejected": -3.383835554122925,
|
| 523 |
+
"logps/chosen": -0.13168923556804657,
|
| 524 |
+
"logps/rejected": -2.653200149536133,
|
| 525 |
+
"loss": 0.1086,
|
| 526 |
+
"nll_loss": 0.10234662145376205,
|
| 527 |
+
"rewards/accuracies": 0.9934999942779541,
|
| 528 |
+
"rewards/chosen": -0.01316892635077238,
|
| 529 |
+
"rewards/margins": 0.2521510720252991,
|
| 530 |
+
"rewards/rejected": -0.2653200030326843,
|
| 531 |
+
"step": 14500
|
| 532 |
+
}
|
| 533 |
+
],
|
| 534 |
+
"logging_steps": 500,
|
| 535 |
+
"max_steps": 14709,
|
| 536 |
+
"num_input_tokens_seen": 0,
|
| 537 |
+
"num_train_epochs": 3,
|
| 538 |
+
"save_steps": 500,
|
| 539 |
+
"total_flos": 0.0,
|
| 540 |
+
"train_batch_size": 1,
|
| 541 |
+
"trial_name": null,
|
| 542 |
+
"trial_params": null
|
| 543 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ae5a922b425098874a0da3804301b0b15fd1825d4ff3710775d4c4c911386ff
|
| 3 |
+
size 5432
|