From c69da8e90f9eeec4ca9755389edc39cf37366e49 Mon Sep 17 00:00:00 2001 From: ModelHub XC Date: Tue, 7 Jul 2026 07:20:13 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96=E9=A1=B9=E7=9B=AE?= =?UTF-8?q?=EF=BC=8C=E7=94=B1ModelHub=20XC=E7=A4=BE=E5=8C=BA=E6=8F=90?= =?UTF-8?q?=E4=BE=9B=E6=A8=A1=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Model: princeton-nlp/gemma-2-9b-it-SimPO Source: Original Platform --- .gitattributes | 49 + README.md | 138 +++ all_results.json | 22 + config.json | 34 + configuration.json | 1 + eval_results.json | 16 + generation_config.json | 8 + model-00001-of-00004.safetensors | 3 + model-00002-of-00004.safetensors | 3 + model-00003-of-00004.safetensors | 3 + model-00004-of-00004.safetensors | 3 + model.safetensors.index.json | 471 ++++++++ special_tokens_map.json | 34 + tokenizer.json | 3 + tokenizer.model | 3 + tokenizer_config.json | 1757 ++++++++++++++++++++++++++++++ train_results.json | 9 + trainer_state.json | 1437 ++++++++++++++++++++++++ training_args.bin | 3 + 19 files changed, 3997 insertions(+) create mode 100644 .gitattributes create mode 100644 README.md create mode 100644 all_results.json create mode 100644 config.json create mode 100644 configuration.json create mode 100644 eval_results.json create mode 100644 generation_config.json create mode 100644 model-00001-of-00004.safetensors create mode 100644 model-00002-of-00004.safetensors create mode 100644 model-00003-of-00004.safetensors create mode 100644 model-00004-of-00004.safetensors create mode 100644 model.safetensors.index.json create mode 100644 special_tokens_map.json create mode 100644 tokenizer.json create mode 100644 tokenizer.model create mode 100644 tokenizer_config.json create mode 100644 train_results.json create mode 100644 trainer_state.json create mode 100644 training_args.bin diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..21b3632 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,49 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bin.* filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zstandard filter=lfs diff=lfs merge=lfs -text +*.tfevents* filter=lfs diff=lfs merge=lfs -text +*.db* filter=lfs diff=lfs merge=lfs -text +*.ark* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.gguf* filter=lfs diff=lfs merge=lfs -text +*.ggml filter=lfs diff=lfs merge=lfs -text +*.llamafile* filter=lfs diff=lfs merge=lfs -text +*.pt2 filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text + +tokenizer.json filter=lfs diff=lfs merge=lfs -text \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..fff74e0 --- /dev/null +++ b/README.md @@ -0,0 +1,138 @@ +--- +base_model: google/gemma-2-9b-it +tags: +- alignment-handbook +- generated_from_trainer +datasets: +- princeton-nlp/gemma2-ultrafeedback-armorm +model-index: +- name: princeton-nlp/gemma-2-9b-it-SimPO + results: [] +license: mit +--- + +# gemma-2-9b-it-SimPO Model Card + +SimPO (Simple Preference Optimization) is an offline preference optimization algorithm designed to enhance the training of large language models (LLMs) with preference optimization datasets. SimPO aligns the reward function with the generation likelihood, eliminating the need for a reference model and incorporating a target reward margin to boost performance. Please refer to our [preprint](https://arxiv.org/pdf/2405.14734) and [github repo](https://github.com/princeton-nlp/SimPO) for more details. + + +## Model Details + +### Model Description + +We fine-tuned [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) on [princeton-nlp/gemma2-ultrafeedback-armorm](https://huggingface.co/datasets/princeton-nlp/gemma2-ultrafeedback-armorm) with the SimPO objective. + +- **Developed by:** Yu Meng, Mengzhou Xia, Danqi Chen +- **Model type:** Causal Language Model +- **License:** gemma +- **Finetuned from model:** [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) + +### Model Sources + + + +- **Repository:** https://github.com/princeton-nlp/SimPO +- **Paper:** https://arxiv.org/pdf/2405.14734 + + +## How to Get Started with the Model +``` +import torch +from transformers import pipeline + +model_id = "princeton-nlp/gemma-2-9b-it-SimPO" + +generator = pipeline( + "text-generation", + model=model_id, + model_kwargs={"torch_dtype": torch.bfloat16}, + device="cuda", +) +outputs = generator([{"role": "user", "content": "What's the difference between llamas and alpacas?"}], + do_sample=False, + eos_token_id=[generator.tokenizer.convert_tokens_to_ids(""), generator.tokenizer.eos_token_id], + max_new_tokens=200) +print(outputs[0]['generated_text']) +``` + +## Training Details + +### Training Data + +We use [princeton-nlp/gemma2-ultrafeedback-armorm](https://huggingface.co/datasets/princeton-nlp/gemma2-ultrafeedback-armorm) as the preference optimization dataset. + +#### Training Hyperparameters + +The hyperparameters used can be found in the [training script](https://github.com/princeton-nlp/SimPO/blob/main/training_configs/gemma-2-9b-it-simpo.yaml). + +#### Speeds, Sizes, Times + +Fine-tuning the [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) on [princeton-nlp/gemma2-ultrafeedback-armorm](https://huggingface.co/datasets/princeton-nlp/gemma2-ultrafeedback-armorm) takes around 100 mins to finish on 8xH100 GPUs. + +## Evaluation Results + + +| models | AE2 LC | AE2 WR | AE2 Length | AH | AH Length | GSM | GSM Length | MMLU | MMLU Length | +|-----------------------------------|:------:|:------:|:----------:|:----:|:---------:|:----:|:----------:|:----:|:-----------:| +| [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) | 51.1 | 38.1 | 1571 | 40.8 | 545 | 87.4 | 395 | 72.7 | 515 | +| [princeton-nlp/gemma-2-9b-it-DPO](https://huggingface.co/princeton-nlp/gemma-2-9b-it-DPO) | 67.8 | 65.4 | 2016 | 58.9 | 717 | 88.5 | 392 | 72.2 | 624 | +| [princeton-nlp/gemma-2-9b-it-SimPO](https://huggingface.co/princeton-nlp/gemma-2-9b-it-SimPO) | 72.4 | 65.9 | 1833 | 59.1 | 693 | 88.0 | 341 | 72.2 | 441 | + + +## Technical Specifications + +### Model Architecture and Objective + +The model architecture is based on [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it). We use the SimPO training objective proposed in our [preprint](https://arxiv.org/pdf/2405.14734). + +#### Hardware + +We used 8xH100 GPUs for model training. + +#### Software + +Training was done using the [alignment-handbook](https://github.com/huggingface/alignment-handbook) library. + +## Citation + +gemma model: +``` +@article{gemma_2024, + title={Gemma}, + url={https://www.kaggle.com/m/3301}, + DOI={10.34740/KAGGLE/M/3301}, + publisher={Kaggle}, + author={Gemma Team}, + year={2024} +} +``` + +SimPO paper: +``` +@article{meng2024simpo, + title={{SimPO}: Simple preference optimization with a reference-free reward}, + author={Meng, Yu and Xia, Mengzhou and Chen, Danqi}, + journal={arXiv preprint arXiv:2405.14734}, + year={2024} +} +``` + +UltraFeedback paper: +``` +@article{cui2023ultrafeedback, + title={{UltraFeedback}: Boosting language models with high-quality feedback}, + author={Cui, Ganqu and Yuan, Lifan and Ding, Ning and Yao, Guanming and Zhu, Wei and Ni, Yuan and Xie, Guotong and Liu, Zhiyuan and Sun, Maosong}, + journal={arXiv preprint arXiv:2310.01377}, + year={2023} +} +``` + +ArmoRM paper: +``` +@article{wang2024interpretable, + title={Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Experts}, + author={Wang, Haoxiang and Xiong, Wei and Xie, Tengyang and Zhao, Han and Zhang, Tong}, + journal={arXiv preprint arXiv:2406.12845}, + year={2024} +} +``` \ No newline at end of file diff --git a/all_results.json b/all_results.json new file mode 100644 index 0000000..fb9c8e1 --- /dev/null +++ b/all_results.json @@ -0,0 +1,22 @@ +{ + "epoch": 0.9989258861439313, + "eval_logits/chosen": -13.382477760314941, + "eval_logits/rejected": -13.246561050415039, + "eval_logps/chosen": -1.8618243932724, + "eval_logps/rejected": -2.3445940017700195, + "eval_loss": 2.7497549057006836, + "eval_rewards/accuracies": 0.7581967115402222, + "eval_rewards/chosen": -18.618242263793945, + "eval_rewards/margins": 4.82769775390625, + "eval_rewards/rejected": -23.445940017700195, + "eval_runtime": 69.1462, + "eval_samples": 1941, + "eval_samples_per_second": 28.071, + "eval_steps_per_second": 0.882, + "total_flos": 0.0, + "train_loss": 3.5518602760889197, + "train_runtime": 6148.2375, + "train_samples": 59569, + "train_samples_per_second": 9.689, + "train_steps_per_second": 0.076 +} \ No newline at end of file diff --git a/config.json b/config.json new file mode 100644 index 0000000..4ff786d --- /dev/null +++ b/config.json @@ -0,0 +1,34 @@ +{ + "_name_or_path": "princeton-nlp/gemma-2-9b-it-SimPO", + "architectures": [ + "Gemma2ForCausalLM" + ], + "attention_bias": false, + "attention_dropout": 0.0, + "attn_logit_softcapping": 50.0, + "bos_token_id": 2, + "cache_implementation": "hybrid", + "eos_token_id": 1, + "final_logit_softcapping": 30.0, + "head_dim": 256, + "hidden_act": "gelu_pytorch_tanh", + "hidden_activation": "gelu_pytorch_tanh", + "hidden_size": 3584, + "initializer_range": 0.02, + "intermediate_size": 14336, + "max_position_embeddings": 8192, + "model_type": "gemma2", + "num_attention_heads": 16, + "num_hidden_layers": 42, + "num_key_value_heads": 8, + "pad_token_id": 0, + "query_pre_attn_scalar": 256, + "rms_norm_eps": 1e-06, + "rope_theta": 10000.0, + "sliding_window": 4096, + "sliding_window_size": 4096, + "torch_dtype": "bfloat16", + "transformers_version": "4.42.4", + "use_cache": true, + "vocab_size": 256000 +} diff --git a/configuration.json b/configuration.json new file mode 100644 index 0000000..bbeeda1 --- /dev/null +++ b/configuration.json @@ -0,0 +1 @@ +{"framework": "pytorch", "task": "text-generation", "allow_remote": true} \ No newline at end of file diff --git a/eval_results.json b/eval_results.json new file mode 100644 index 0000000..27135c6 --- /dev/null +++ b/eval_results.json @@ -0,0 +1,16 @@ +{ + "epoch": 0.9989258861439313, + "eval_logits/chosen": -13.382477760314941, + "eval_logits/rejected": -13.246561050415039, + "eval_logps/chosen": -1.8618243932724, + "eval_logps/rejected": -2.3445940017700195, + "eval_loss": 2.7497549057006836, + "eval_rewards/accuracies": 0.7581967115402222, + "eval_rewards/chosen": -18.618242263793945, + "eval_rewards/margins": 4.82769775390625, + "eval_rewards/rejected": -23.445940017700195, + "eval_runtime": 69.1462, + "eval_samples": 1941, + "eval_samples_per_second": 28.071, + "eval_steps_per_second": 0.882 +} \ No newline at end of file diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..d9c3c77 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,8 @@ +{ + "_from_model_config": true, + "bos_token_id": 2, + "cache_implementation": "hybrid", + "eos_token_id": [1, 107], + "pad_token_id": 0, + "transformers_version": "4.42.4" +} diff --git a/model-00001-of-00004.safetensors b/model-00001-of-00004.safetensors new file mode 100644 index 0000000..927bc96 --- /dev/null +++ b/model-00001-of-00004.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a013695a0890be87cca5a3adb49547ffa909b14ead512002cd72106b9a6e2622 +size 4903351912 diff --git a/model-00002-of-00004.safetensors b/model-00002-of-00004.safetensors new file mode 100644 index 0000000..6a01130 --- /dev/null +++ b/model-00002-of-00004.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3f340da7e3a6f7bc0d6611d852f795bea0dba334f0204ea9d1df1dc2a4362cf1 +size 4947570872 diff --git a/model-00003-of-00004.safetensors b/model-00003-of-00004.safetensors new file mode 100644 index 0000000..d6a628a --- /dev/null +++ b/model-00003-of-00004.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3ad5697776ba58c1f956d94382dd441055a3268db05bfc491422377cefbc875a +size 4962221464 diff --git a/model-00004-of-00004.safetensors b/model-00004-of-00004.safetensors new file mode 100644 index 0000000..4e0de25 --- /dev/null +++ b/model-00004-of-00004.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:344c4bb51c8b32bb20005c785098f554b076890133c02b05535b373e4108c300 +size 3670322200 diff --git a/model.safetensors.index.json b/model.safetensors.index.json new file mode 100644 index 0000000..43bf0e4 --- /dev/null +++ b/model.safetensors.index.json @@ -0,0 +1,471 @@ +{ + "metadata": { + "total_size": 18483411968 + }, + "weight_map": { + "model.embed_tokens.weight": "model-00001-of-00004.safetensors", + "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.0.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.0.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.1.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.1.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.10.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.10.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.11.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.11.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.12.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.12.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.13.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.13.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.14.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.14.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.15.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.15.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.16.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.16.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.17.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.17.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.18.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.18.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.19.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.19.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.2.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.2.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.20.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.20.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.21.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.21.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.22.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.22.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.23.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.23.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.24.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.24.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.25.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.25.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.26.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.26.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.27.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.27.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.28.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.28.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.29.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.29.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.3.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.3.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.30.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.30.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.input_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.31.mlp.down_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.post_attention_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.31.post_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.31.pre_feedforward_layernorm.weight": "model-00003-of-00004.safetensors", + "model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.32.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.32.mlp.gate_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.mlp.up_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.32.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.32.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.32.self_attn.k_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.self_attn.o_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.self_attn.q_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.32.self_attn.v_proj.weight": "model-00003-of-00004.safetensors", + "model.layers.33.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.33.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.33.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.33.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.33.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.33.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.34.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.34.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.34.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.34.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.34.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.35.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.35.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.35.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.35.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.35.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.36.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.36.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.36.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.36.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.36.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.37.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.37.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.37.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.37.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.37.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.38.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.38.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.38.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.38.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.38.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.39.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.39.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.39.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.39.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.39.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.4.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.4.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.40.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.40.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.40.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.40.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.40.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.40.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.input_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.41.mlp.down_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.mlp.gate_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.mlp.up_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.post_attention_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.41.post_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.41.pre_feedforward_layernorm.weight": "model-00004-of-00004.safetensors", + "model.layers.41.self_attn.k_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.self_attn.o_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.self_attn.q_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.41.self_attn.v_proj.weight": "model-00004-of-00004.safetensors", + "model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.5.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.5.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.6.post_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.6.pre_feedforward_layernorm.weight": "model-00001-of-00004.safetensors", + "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.7.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.7.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.7.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors", + "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.8.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.8.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.9.post_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.9.pre_feedforward_layernorm.weight": "model-00002-of-00004.safetensors", + "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors", + "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors", + "model.norm.weight": "model-00004-of-00004.safetensors" + } +} diff --git a/special_tokens_map.json b/special_tokens_map.json new file mode 100644 index 0000000..8d6368f --- /dev/null +++ b/special_tokens_map.json @@ -0,0 +1,34 @@ +{ + "additional_special_tokens": [ + "", + "" + ], + "bos_token": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "eos_token": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "pad_token": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + }, + "unk_token": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false + } +} diff --git a/tokenizer.json b/tokenizer.json new file mode 100644 index 0000000..5771f48 --- /dev/null +++ b/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7da53ca29fb16f6b2489482fc0bc6a394162cdab14d12764a1755ebc583fea79 +size 17518525 diff --git a/tokenizer.model b/tokenizer.model new file mode 100644 index 0000000..796efe9 --- /dev/null +++ b/tokenizer.model @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2 +size 4241003 diff --git a/tokenizer_config.json b/tokenizer_config.json new file mode 100644 index 0000000..fd99808 --- /dev/null +++ b/tokenizer_config.json @@ -0,0 +1,1757 @@ +{ + "add_bos_token": true, + "add_eos_token": false, + "added_tokens_decoder": { + "0": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "1": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "2": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "3": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "4": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "5": { + "content": "<2mass>", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "6": { + "content": "[@BOS@]", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "7": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "8": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "9": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "10": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "11": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "12": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "13": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "14": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "15": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "16": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "17": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "18": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "19": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "20": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "21": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "22": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "23": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "24": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "25": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "26": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "27": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "28": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "29": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "30": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "31": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "32": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "33": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "34": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "35": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "36": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "37": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "38": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "39": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "40": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "41": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "42": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "43": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "44": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "45": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "46": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "47": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "48": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "49": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "50": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "51": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "52": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "53": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "54": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "55": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "56": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "57": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "58": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "59": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "60": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "61": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "62": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "63": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "64": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "65": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "66": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "67": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "68": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "69": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "70": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "71": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "72": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "73": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "74": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "75": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "76": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "77": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "78": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "79": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "80": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "81": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "82": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "83": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "84": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "85": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "86": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "87": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "88": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "89": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "90": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "91": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "92": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "93": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "94": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "95": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "96": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "97": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "98": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "99": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "100": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "101": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "102": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "103": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "104": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "105": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "106": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "107": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": true + }, + "108": { + "content": "\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "109": { + "content": "\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "110": { + "content": "\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "111": { + "content": "\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "112": { + "content": "\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "113": { + "content": "\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "114": { + "content": "\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "115": { + "content": "\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "116": { + "content": "\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "117": { + "content": "\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "118": { + "content": "\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "119": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "120": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "121": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "122": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "123": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "124": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "125": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "126": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "127": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "128": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "129": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "130": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "131": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "132": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "133": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "134": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "135": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "136": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "137": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "138": { + "content": "\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "139": { + "content": "▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "140": { + "content": "▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "141": { + "content": "▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "142": { + "content": "▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "143": { + "content": "▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "144": { + "content": "▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "145": { + "content": "▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "146": { + "content": "▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "147": { + "content": "▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "148": { + "content": "▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "149": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "150": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "151": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "152": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "153": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "154": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "155": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "156": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "157": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "158": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "159": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "160": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "161": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "162": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "163": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "164": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "165": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "166": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "167": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "168": { + "content": "▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "169": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "170": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "172": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "173": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "174": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "175": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "171": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "176": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "177": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "178": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "179": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "180": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "181": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "182": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "183": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "184": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "185": { + "content": "

", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "186": { + "content": "

", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "187": { + "content": "

", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "188": { + "content": "

", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "189": { + "content": "

", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "190": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "191": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "192": { + "content": "
", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "193": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "194": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "195": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "196": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "197": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "198": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "199": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "200": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "201": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "202": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "203": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "204": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "205": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "206": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "207": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "208": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "209": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "210": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "211": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "212": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "213": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "214": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "215": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + }, + "216": { + "content": "", + "lstrip": false, + "normalized": false, + "rstrip": false, + "single_word": false, + "special": false + } + }, + "additional_special_tokens": [ + "", + "" + ], + "bos_token": "", + "chat_template": "{{ bos_token }}{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] | trim + '\n\n' %}{% set messages = messages[1:] %}{% else %}{% set system_message = '' %}{% endif %}{% for message in messages %}{% if loop.index0 == 0 %}{% set content = system_message + message['content'] %}{% else %}{% set content = message['content'] %}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '' + role + '\n' + content | trim + '\n' }}{% endfor %}{% if add_generation_prompt %}{{'model\n'}}{% endif %}", + "clean_up_tokenization_spaces": false, + "eos_token": "", + "model_max_length": 2048, + "pad_token": "", + "sp_model_kwargs": {}, + "spaces_between_special_tokens": false, + "tokenizer_class": "GemmaTokenizer", + "unk_token": "", + "use_default_system_prompt": false +} diff --git a/train_results.json b/train_results.json new file mode 100644 index 0000000..66173eb --- /dev/null +++ b/train_results.json @@ -0,0 +1,9 @@ +{ + "epoch": 0.9989258861439313, + "total_flos": 0.0, + "train_loss": 3.5518602760889197, + "train_runtime": 6148.2375, + "train_samples": 59569, + "train_samples_per_second": 9.689, + "train_steps_per_second": 0.076 +} \ No newline at end of file diff --git a/trainer_state.json b/trainer_state.json new file mode 100644 index 0000000..d746f99 --- /dev/null +++ b/trainer_state.json @@ -0,0 +1,1437 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.9989258861439313, + "eval_steps": 100000, + "global_step": 465, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.010741138560687433, + "grad_norm": 94.5147817778946, + "learning_rate": 8.51063829787234e-08, + "logits/chosen": -10.583702087402344, + "logits/rejected": -10.455877304077148, + "logps/chosen": -0.9049979448318481, + "logps/rejected": -0.8784100413322449, + "loss": 6.1451, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -9.04997730255127, + "rewards/margins": -0.2658771872520447, + "rewards/rejected": -8.784101486206055, + "step": 5 + }, + { + "epoch": 0.021482277121374866, + "grad_norm": 128.5515421485228, + "learning_rate": 1.702127659574468e-07, + "logits/chosen": -10.710015296936035, + "logits/rejected": -10.85377311706543, + "logps/chosen": -1.0046945810317993, + "logps/rejected": -0.8850045204162598, + "loss": 5.8491, + "rewards/accuracies": 0.512499988079071, + "rewards/chosen": -10.04694652557373, + "rewards/margins": -1.196901559829712, + "rewards/rejected": -8.850046157836914, + "step": 10 + }, + { + "epoch": 0.0322234156820623, + "grad_norm": 58.802331595913145, + "learning_rate": 2.553191489361702e-07, + "logits/chosen": -10.312850952148438, + "logits/rejected": -10.239133834838867, + "logps/chosen": -1.0889472961425781, + "logps/rejected": -1.1543949842453003, + "loss": 6.2505, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -10.889472961425781, + "rewards/margins": 0.6544777154922485, + "rewards/rejected": -11.543951034545898, + "step": 15 + }, + { + "epoch": 0.04296455424274973, + "grad_norm": 151.21348799898024, + "learning_rate": 3.404255319148936e-07, + "logits/chosen": -9.954164505004883, + "logits/rejected": -10.053568840026855, + "logps/chosen": -0.9611791372299194, + "logps/rejected": -1.1332345008850098, + "loss": 5.5619, + "rewards/accuracies": 0.44999998807907104, + "rewards/chosen": -9.611791610717773, + "rewards/margins": 1.7205528020858765, + "rewards/rejected": -11.332345008850098, + "step": 20 + }, + { + "epoch": 0.05370569280343716, + "grad_norm": 99.92865956439873, + "learning_rate": 4.25531914893617e-07, + "logits/chosen": -10.326103210449219, + "logits/rejected": -10.055009841918945, + "logps/chosen": -0.8260948061943054, + "logps/rejected": -1.1549828052520752, + "loss": 5.2635, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": -8.260948181152344, + "rewards/margins": 3.2888808250427246, + "rewards/rejected": -11.54982852935791, + "step": 25 + }, + { + "epoch": 0.0644468313641246, + "grad_norm": 81.6600880763309, + "learning_rate": 5.106382978723404e-07, + "logits/chosen": -9.319940567016602, + "logits/rejected": -9.13192081451416, + "logps/chosen": -0.6618553996086121, + "logps/rejected": -0.6553267240524292, + "loss": 5.0518, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -6.61855411529541, + "rewards/margins": -0.06528709828853607, + "rewards/rejected": -6.553267002105713, + "step": 30 + }, + { + "epoch": 0.07518796992481203, + "grad_norm": 92.79131753018717, + "learning_rate": 5.957446808510638e-07, + "logits/chosen": -8.877812385559082, + "logits/rejected": -8.929550170898438, + "logps/chosen": -0.717892050743103, + "logps/rejected": -0.6935927867889404, + "loss": 5.224, + "rewards/accuracies": 0.512499988079071, + "rewards/chosen": -7.178920745849609, + "rewards/margins": -0.24299363791942596, + "rewards/rejected": -6.935927391052246, + "step": 35 + }, + { + "epoch": 0.08592910848549946, + "grad_norm": 89.40050001716304, + "learning_rate": 6.808510638297872e-07, + "logits/chosen": -7.988096714019775, + "logits/rejected": -7.907191276550293, + "logps/chosen": -0.7402302622795105, + "logps/rejected": -0.7434382438659668, + "loss": 4.8885, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -7.4023027420043945, + "rewards/margins": 0.032080501317977905, + "rewards/rejected": -7.434383392333984, + "step": 40 + }, + { + "epoch": 0.0966702470461869, + "grad_norm": 59.92644904113339, + "learning_rate": 7.659574468085107e-07, + "logits/chosen": -8.71805477142334, + "logits/rejected": -8.232014656066895, + "logps/chosen": -0.5317873954772949, + "logps/rejected": -0.6050616502761841, + "loss": 4.5879, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -5.317873477935791, + "rewards/margins": 0.7327424883842468, + "rewards/rejected": -6.0506157875061035, + "step": 45 + }, + { + "epoch": 0.10741138560687433, + "grad_norm": 60.25452206880678, + "learning_rate": 7.998983280184396e-07, + "logits/chosen": -8.83049488067627, + "logits/rejected": -8.585375785827637, + "logps/chosen": -0.5144228339195251, + "logps/rejected": -0.5809676647186279, + "loss": 4.6549, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -5.144228458404541, + "rewards/margins": 0.6654484868049622, + "rewards/rejected": -5.8096771240234375, + "step": 50 + }, + { + "epoch": 0.11815252416756176, + "grad_norm": 46.997605934809734, + "learning_rate": 7.992771864078597e-07, + "logits/chosen": -8.163946151733398, + "logits/rejected": -8.180994033813477, + "logps/chosen": -0.5956984758377075, + "logps/rejected": -0.7000880837440491, + "loss": 4.6606, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -5.956984519958496, + "rewards/margins": 1.0438958406448364, + "rewards/rejected": -7.000881195068359, + "step": 55 + }, + { + "epoch": 0.1288936627282492, + "grad_norm": 59.25357465703395, + "learning_rate": 7.980922636120897e-07, + "logits/chosen": -8.718216896057129, + "logits/rejected": -8.35698127746582, + "logps/chosen": -0.5706155896186829, + "logps/rejected": -0.6969493627548218, + "loss": 4.4885, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -5.706155776977539, + "rewards/margins": 1.2633379697799683, + "rewards/rejected": -6.969493865966797, + "step": 60 + }, + { + "epoch": 0.13963480128893663, + "grad_norm": 50.79780612404402, + "learning_rate": 7.963452327474534e-07, + "logits/chosen": -9.234804153442383, + "logits/rejected": -9.1095609664917, + "logps/chosen": -0.6090906858444214, + "logps/rejected": -0.7208055257797241, + "loss": 4.666, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -6.090908050537109, + "rewards/margins": 1.1171473264694214, + "rewards/rejected": -7.208055019378662, + "step": 65 + }, + { + "epoch": 0.15037593984962405, + "grad_norm": 56.36564641791114, + "learning_rate": 7.940385606293987e-07, + "logits/chosen": -8.946883201599121, + "logits/rejected": -8.716778755187988, + "logps/chosen": -0.6818052530288696, + "logps/rejected": -0.7961267828941345, + "loss": 4.577, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -6.818052768707275, + "rewards/margins": 1.1432150602340698, + "rewards/rejected": -7.961267948150635, + "step": 70 + }, + { + "epoch": 0.1611170784103115, + "grad_norm": 61.76930510948969, + "learning_rate": 7.911755042893434e-07, + "logits/chosen": -9.067525863647461, + "logits/rejected": -8.9346923828125, + "logps/chosen": -0.6832990646362305, + "logps/rejected": -0.7763570547103882, + "loss": 4.4179, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -6.8329901695251465, + "rewards/margins": 0.9305804371833801, + "rewards/rejected": -7.763571262359619, + "step": 75 + }, + { + "epoch": 0.17185821697099893, + "grad_norm": 63.602633074853536, + "learning_rate": 7.877601063757321e-07, + "logits/chosen": -9.461370468139648, + "logits/rejected": -8.981520652770996, + "logps/chosen": -0.6881433129310608, + "logps/rejected": -0.8508684039115906, + "loss": 4.3763, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -6.881433010101318, + "rewards/margins": 1.6272509098052979, + "rewards/rejected": -8.508684158325195, + "step": 80 + }, + { + "epoch": 0.18259935553168635, + "grad_norm": 50.67613033462041, + "learning_rate": 7.837971894457989e-07, + "logits/chosen": -9.557887077331543, + "logits/rejected": -9.17081069946289, + "logps/chosen": -0.6830392479896545, + "logps/rejected": -0.799291729927063, + "loss": 4.6499, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -6.830392360687256, + "rewards/margins": 1.1625245809555054, + "rewards/rejected": -7.992917060852051, + "step": 85 + }, + { + "epoch": 0.1933404940923738, + "grad_norm": 57.58777625940833, + "learning_rate": 7.792923491560942e-07, + "logits/chosen": -8.579484939575195, + "logits/rejected": -8.546136856079102, + "logps/chosen": -0.6667743921279907, + "logps/rejected": -0.7407978177070618, + "loss": 4.4492, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -6.667743682861328, + "rewards/margins": 0.7402342557907104, + "rewards/rejected": -7.407977104187012, + "step": 90 + }, + { + "epoch": 0.20408163265306123, + "grad_norm": 68.91179001810359, + "learning_rate": 7.742519463613926e-07, + "logits/chosen": -9.336307525634766, + "logits/rejected": -9.128133773803711, + "logps/chosen": -0.706219494342804, + "logps/rejected": -0.7757526636123657, + "loss": 4.2763, + "rewards/accuracies": 0.5, + "rewards/chosen": -7.062193870544434, + "rewards/margins": 0.6953321099281311, + "rewards/rejected": -7.7575273513793945, + "step": 95 + }, + { + "epoch": 0.21482277121374865, + "grad_norm": 140.43504047927686, + "learning_rate": 7.68683098133138e-07, + "logits/chosen": -8.939419746398926, + "logits/rejected": -8.681028366088867, + "logps/chosen": -0.7093919515609741, + "logps/rejected": -0.8932901620864868, + "loss": 4.4002, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -7.093919277191162, + "rewards/margins": 1.838982343673706, + "rewards/rejected": -8.932901382446289, + "step": 100 + }, + { + "epoch": 0.22556390977443608, + "grad_norm": 105.52949175741638, + "learning_rate": 7.625936677101051e-07, + "logits/chosen": -8.601816177368164, + "logits/rejected": -8.625459671020508, + "logps/chosen": -0.8767679333686829, + "logps/rejected": -0.8515819311141968, + "loss": 4.4644, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -8.767679214477539, + "rewards/margins": -0.25185948610305786, + "rewards/rejected": -8.51581859588623, + "step": 105 + }, + { + "epoch": 0.23630504833512353, + "grad_norm": 56.44051818244315, + "learning_rate": 7.559922533954731e-07, + "logits/chosen": -9.58240795135498, + "logits/rejected": -9.501542091369629, + "logps/chosen": -0.7655607461929321, + "logps/rejected": -0.8916142582893372, + "loss": 4.2797, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -7.6556077003479, + "rewards/margins": 1.2605348825454712, + "rewards/rejected": -8.916143417358398, + "step": 110 + }, + { + "epoch": 0.24704618689581095, + "grad_norm": 68.9497539150874, + "learning_rate": 7.488881764159808e-07, + "logits/chosen": -9.756335258483887, + "logits/rejected": -9.543218612670898, + "logps/chosen": -0.7038711309432983, + "logps/rejected": -0.7986757159233093, + "loss": 4.154, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -7.0387115478515625, + "rewards/margins": 0.9480463862419128, + "rewards/rejected": -7.986758232116699, + "step": 115 + }, + { + "epoch": 0.2577873254564984, + "grad_norm": 90.20504733911939, + "learning_rate": 7.412914677603135e-07, + "logits/chosen": -9.883420944213867, + "logits/rejected": -9.735390663146973, + "logps/chosen": -0.9017173647880554, + "logps/rejected": -0.9996153116226196, + "loss": 4.2168, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -9.017173767089844, + "rewards/margins": 0.9789786338806152, + "rewards/rejected": -9.996152877807617, + "step": 120 + }, + { + "epoch": 0.26852846401718583, + "grad_norm": 63.31580116626401, + "learning_rate": 7.332128540153017e-07, + "logits/chosen": -10.71928596496582, + "logits/rejected": -10.555776596069336, + "logps/chosen": -0.788918673992157, + "logps/rejected": -0.9437187910079956, + "loss": 4.1085, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -7.889185905456543, + "rewards/margins": 1.5480016469955444, + "rewards/rejected": -9.437189102172852, + "step": 125 + }, + { + "epoch": 0.27926960257787325, + "grad_norm": 81.18347953083857, + "learning_rate": 7.246637422199322e-07, + "logits/chosen": -10.676037788391113, + "logits/rejected": -10.630210876464844, + "logps/chosen": -0.8381876945495605, + "logps/rejected": -1.0221493244171143, + "loss": 4.0857, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -8.381875991821289, + "rewards/margins": 1.8396151065826416, + "rewards/rejected": -10.221491813659668, + "step": 130 + }, + { + "epoch": 0.2900107411385607, + "grad_norm": 80.90445716094146, + "learning_rate": 7.156562037585574e-07, + "logits/chosen": -11.714326858520508, + "logits/rejected": -11.1636323928833, + "logps/chosen": -0.8452903628349304, + "logps/rejected": -1.1077954769134521, + "loss": 3.9406, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -8.452905654907227, + "rewards/margins": 2.6250510215759277, + "rewards/rejected": -11.07795524597168, + "step": 135 + }, + { + "epoch": 0.3007518796992481, + "grad_norm": 88.8085364984583, + "learning_rate": 7.062029573160467e-07, + "logits/chosen": -11.935297012329102, + "logits/rejected": -11.792046546936035, + "logps/chosen": -0.9109989404678345, + "logps/rejected": -1.1446388959884644, + "loss": 3.6921, + "rewards/accuracies": 0.8125, + "rewards/chosen": -9.109989166259766, + "rewards/margins": 2.3363993167877197, + "rewards/rejected": -11.446390151977539, + "step": 140 + }, + { + "epoch": 0.31149301825993553, + "grad_norm": 110.78828702618637, + "learning_rate": 6.963173509189455e-07, + "logits/chosen": -13.552042007446289, + "logits/rejected": -13.324705123901367, + "logps/chosen": -1.0502710342407227, + "logps/rejected": -1.2693006992340088, + "loss": 3.8236, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -10.502711296081543, + "rewards/margins": 2.190295696258545, + "rewards/rejected": -12.69300651550293, + "step": 145 + }, + { + "epoch": 0.322234156820623, + "grad_norm": 111.15417892636452, + "learning_rate": 6.860133430880024e-07, + "logits/chosen": -14.586761474609375, + "logits/rejected": -14.23077392578125, + "logps/chosen": -1.2444875240325928, + "logps/rejected": -1.4712624549865723, + "loss": 3.7527, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -12.444875717163086, + "rewards/margins": 2.2677478790283203, + "rewards/rejected": -14.712623596191406, + "step": 150 + }, + { + "epoch": 0.33297529538131043, + "grad_norm": 119.28100769496392, + "learning_rate": 6.753054831286747e-07, + "logits/chosen": -14.823234558105469, + "logits/rejected": -14.770757675170898, + "logps/chosen": -1.3944904804229736, + "logps/rejected": -1.6155385971069336, + "loss": 3.893, + "rewards/accuracies": 0.6875, + "rewards/chosen": -13.944903373718262, + "rewards/margins": 2.2104804515838623, + "rewards/rejected": -16.155384063720703, + "step": 155 + }, + { + "epoch": 0.34371643394199786, + "grad_norm": 103.61310969210844, + "learning_rate": 6.642088905874433e-07, + "logits/chosen": -14.195696830749512, + "logits/rejected": -14.07690143585205, + "logps/chosen": -1.3757129907608032, + "logps/rejected": -1.5891997814178467, + "loss": 3.7363, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -13.75713062286377, + "rewards/margins": 2.134868860244751, + "rewards/rejected": -15.891998291015625, + "step": 160 + }, + { + "epoch": 0.3544575725026853, + "grad_norm": 95.46872943421629, + "learning_rate": 6.527392339029455e-07, + "logits/chosen": -14.401777267456055, + "logits/rejected": -14.322749137878418, + "logps/chosen": -1.2530758380889893, + "logps/rejected": -1.5336121320724487, + "loss": 3.4763, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -12.530759811401367, + "rewards/margins": 2.8053627014160156, + "rewards/rejected": -15.336122512817383, + "step": 165 + }, + { + "epoch": 0.3651987110633727, + "grad_norm": 119.23452912777815, + "learning_rate": 6.409127082820689e-07, + "logits/chosen": -14.566454887390137, + "logits/rejected": -14.484842300415039, + "logps/chosen": -1.531166434288025, + "logps/rejected": -1.7996248006820679, + "loss": 3.6554, + "rewards/accuracies": 0.6875, + "rewards/chosen": -15.311663627624512, + "rewards/margins": 2.6845829486846924, + "rewards/rejected": -17.996248245239258, + "step": 170 + }, + { + "epoch": 0.37593984962406013, + "grad_norm": 123.93953756068933, + "learning_rate": 6.287460128322457e-07, + "logits/chosen": -14.157377243041992, + "logits/rejected": -14.0371675491333, + "logps/chosen": -1.5080561637878418, + "logps/rejected": -1.9248558282852173, + "loss": 3.4374, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -15.080561637878418, + "rewards/margins": 4.167994022369385, + "rewards/rejected": -19.24855613708496, + "step": 175 + }, + { + "epoch": 0.3866809881847476, + "grad_norm": 223.710665367037, + "learning_rate": 6.16256326982239e-07, + "logits/chosen": -16.03777313232422, + "logits/rejected": -16.100345611572266, + "logps/chosen": -1.5300877094268799, + "logps/rejected": -1.8575336933135986, + "loss": 3.4606, + "rewards/accuracies": 0.6875, + "rewards/chosen": -15.300875663757324, + "rewards/margins": 3.2744598388671875, + "rewards/rejected": -18.575336456298828, + "step": 180 + }, + { + "epoch": 0.39742212674543503, + "grad_norm": 294.4348098673441, + "learning_rate": 6.034612862247114e-07, + "logits/chosen": -14.142799377441406, + "logits/rejected": -13.796422958374023, + "logps/chosen": -1.5025275945663452, + "logps/rejected": -1.7810484170913696, + "loss": 3.1185, + "rewards/accuracies": 0.6875, + "rewards/chosen": -15.025274276733398, + "rewards/margins": 2.7852089405059814, + "rewards/rejected": -17.810483932495117, + "step": 185 + }, + { + "epoch": 0.40816326530612246, + "grad_norm": 112.37773385751002, + "learning_rate": 5.903789572148295e-07, + "logits/chosen": -14.8009614944458, + "logits/rejected": -14.250249862670898, + "logps/chosen": -1.5931546688079834, + "logps/rejected": -2.022378444671631, + "loss": 3.3847, + "rewards/accuracies": 0.75, + "rewards/chosen": -15.931546211242676, + "rewards/margins": 4.292238712310791, + "rewards/rejected": -20.223783493041992, + "step": 190 + }, + { + "epoch": 0.4189044038668099, + "grad_norm": 125.77761635765717, + "learning_rate": 5.770278122600662e-07, + "logits/chosen": -14.832977294921875, + "logits/rejected": -14.608530044555664, + "logps/chosen": -1.6177564859390259, + "logps/rejected": -1.9727256298065186, + "loss": 3.6009, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -16.177562713623047, + "rewards/margins": 3.549692153930664, + "rewards/rejected": -19.72725486755371, + "step": 195 + }, + { + "epoch": 0.4296455424274973, + "grad_norm": 101.70153964682511, + "learning_rate": 5.634267032372192e-07, + "logits/chosen": -14.803668022155762, + "logits/rejected": -14.786203384399414, + "logps/chosen": -1.6423594951629639, + "logps/rejected": -1.9881032705307007, + "loss": 3.3904, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -16.423595428466797, + "rewards/margins": 3.457437515258789, + "rewards/rejected": -19.881032943725586, + "step": 200 + }, + { + "epoch": 0.44038668098818473, + "grad_norm": 123.22102599092351, + "learning_rate": 5.495948349734758e-07, + "logits/chosen": -14.582061767578125, + "logits/rejected": -14.51270580291748, + "logps/chosen": -1.747982382774353, + "logps/rejected": -2.051506757736206, + "loss": 3.1521, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -17.47982406616211, + "rewards/margins": 3.035243034362793, + "rewards/rejected": -20.51506805419922, + "step": 205 + }, + { + "epoch": 0.45112781954887216, + "grad_norm": 106.53206062736002, + "learning_rate": 5.355517381291105e-07, + "logits/chosen": -15.856142044067383, + "logits/rejected": -15.048059463500977, + "logps/chosen": -1.9086406230926514, + "logps/rejected": -2.3940463066101074, + "loss": 3.1856, + "rewards/accuracies": 0.75, + "rewards/chosen": -19.086406707763672, + "rewards/margins": 4.854057312011719, + "rewards/rejected": -23.940462112426758, + "step": 210 + }, + { + "epoch": 0.46186895810955964, + "grad_norm": 135.01895835142213, + "learning_rate": 5.21317241620105e-07, + "logits/chosen": -17.223520278930664, + "logits/rejected": -16.822795867919922, + "logps/chosen": -1.9180253744125366, + "logps/rejected": -2.2895708084106445, + "loss": 3.4171, + "rewards/accuracies": 0.75, + "rewards/chosen": -19.180253982543945, + "rewards/margins": 3.7154533863067627, + "rewards/rejected": -22.895706176757812, + "step": 215 + }, + { + "epoch": 0.47261009667024706, + "grad_norm": 116.88786548811646, + "learning_rate": 5.069114446196291e-07, + "logits/chosen": -14.405430793762207, + "logits/rejected": -14.03125, + "logps/chosen": -1.8195463418960571, + "logps/rejected": -2.373373508453369, + "loss": 3.1087, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -18.195463180541992, + "rewards/margins": 5.538268566131592, + "rewards/rejected": -23.73373031616211, + "step": 220 + }, + { + "epoch": 0.4833512352309345, + "grad_norm": 117.33800643418842, + "learning_rate": 4.923546881779183e-07, + "logits/chosen": -15.34239387512207, + "logits/rejected": -15.118896484375, + "logps/chosen": -1.5427398681640625, + "logps/rejected": -1.9821481704711914, + "loss": 3.0885, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -15.427398681640625, + "rewards/margins": 4.394083023071289, + "rewards/rejected": -19.821481704711914, + "step": 225 + }, + { + "epoch": 0.4940923737916219, + "grad_norm": 116.49865117484065, + "learning_rate": 4.776675265006186e-07, + "logits/chosen": -14.630195617675781, + "logits/rejected": -14.59937858581543, + "logps/chosen": -1.6563146114349365, + "logps/rejected": -2.0928232669830322, + "loss": 3.2032, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -16.56314468383789, + "rewards/margins": 4.365086555480957, + "rewards/rejected": -20.928232192993164, + "step": 230 + }, + { + "epoch": 0.5048335123523093, + "grad_norm": 203.29377037587201, + "learning_rate": 4.62870697926156e-07, + "logits/chosen": -14.8600435256958, + "logits/rejected": -15.338279724121094, + "logps/chosen": -1.7831714153289795, + "logps/rejected": -2.145383358001709, + "loss": 3.2633, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -17.831714630126953, + "rewards/margins": 3.622117519378662, + "rewards/rejected": -21.45383071899414, + "step": 235 + }, + { + "epoch": 0.5155746509129968, + "grad_norm": 116.16395762640381, + "learning_rate": 4.479850956431092e-07, + "logits/chosen": -14.476922988891602, + "logits/rejected": -14.742956161499023, + "logps/chosen": -1.6810489892959595, + "logps/rejected": -2.0232789516448975, + "loss": 3.0869, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -16.810489654541016, + "rewards/margins": 3.4223015308380127, + "rewards/rejected": -20.232791900634766, + "step": 240 + }, + { + "epoch": 0.5263157894736842, + "grad_norm": 98.59837238595348, + "learning_rate": 4.33031738188933e-07, + "logits/chosen": -15.394210815429688, + "logits/rejected": -14.864255905151367, + "logps/chosen": -1.6588356494903564, + "logps/rejected": -2.0340933799743652, + "loss": 3.2303, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -16.58835792541504, + "rewards/margins": 3.752579927444458, + "rewards/rejected": -20.3409366607666, + "step": 245 + }, + { + "epoch": 0.5370569280343717, + "grad_norm": 122.58746074195308, + "learning_rate": 4.180317397716889e-07, + "logits/chosen": -15.588345527648926, + "logits/rejected": -15.222723007202148, + "logps/chosen": -1.6328115463256836, + "logps/rejected": -2.2505877017974854, + "loss": 3.0874, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -16.328113555908203, + "rewards/margins": 6.177763938903809, + "rewards/rejected": -22.505878448486328, + "step": 250 + }, + { + "epoch": 0.547798066595059, + "grad_norm": 120.49234048427142, + "learning_rate": 4.030062804566888e-07, + "logits/chosen": -15.462881088256836, + "logits/rejected": -15.3878173828125, + "logps/chosen": -1.6617505550384521, + "logps/rejected": -1.9518375396728516, + "loss": 3.0471, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -16.61750602722168, + "rewards/margins": 2.9008688926696777, + "rewards/rejected": -19.518375396728516, + "step": 255 + }, + { + "epoch": 0.5585392051557465, + "grad_norm": 102.36282696415171, + "learning_rate": 3.8797657626014614e-07, + "logits/chosen": -15.693799018859863, + "logits/rejected": -15.475125312805176, + "logps/chosen": -1.7303409576416016, + "logps/rejected": -2.111295223236084, + "loss": 3.1486, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -17.303409576416016, + "rewards/margins": 3.809544801712036, + "rewards/rejected": -21.112953186035156, + "step": 260 + }, + { + "epoch": 0.569280343716434, + "grad_norm": 120.30352416855824, + "learning_rate": 3.729638491920669e-07, + "logits/chosen": -14.018827438354492, + "logits/rejected": -14.128240585327148, + "logps/chosen": -1.5650581121444702, + "logps/rejected": -1.8844165802001953, + "loss": 3.0723, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -15.650581359863281, + "rewards/margins": 3.193586826324463, + "rewards/rejected": -18.84417152404785, + "step": 265 + }, + { + "epoch": 0.5800214822771214, + "grad_norm": 91.11946128351651, + "learning_rate": 3.5798929729067464e-07, + "logits/chosen": -15.980966567993164, + "logits/rejected": -15.59577465057373, + "logps/chosen": -1.759478211402893, + "logps/rejected": -2.1487960815429688, + "loss": 2.7323, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": -17.59478187561035, + "rewards/margins": 3.8931777477264404, + "rewards/rejected": -21.487960815429688, + "step": 270 + }, + { + "epoch": 0.5907626208378088, + "grad_norm": 111.96074414088407, + "learning_rate": 3.4307406469068595e-07, + "logits/chosen": -15.691810607910156, + "logits/rejected": -15.631698608398438, + "logps/chosen": -1.7383606433868408, + "logps/rejected": -2.2635440826416016, + "loss": 2.9053, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -17.38360595703125, + "rewards/margins": 5.251835823059082, + "rewards/rejected": -22.635440826416016, + "step": 275 + }, + { + "epoch": 0.6015037593984962, + "grad_norm": 122.95732420434838, + "learning_rate": 3.282392117676968e-07, + "logits/chosen": -15.389913558959961, + "logits/rejected": -15.556841850280762, + "logps/chosen": -1.9042613506317139, + "logps/rejected": -2.476677417755127, + "loss": 2.9112, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": -19.04261589050293, + "rewards/margins": 5.72415828704834, + "rewards/rejected": -24.766775131225586, + "step": 280 + }, + { + "epoch": 0.6122448979591837, + "grad_norm": 104.08928006990811, + "learning_rate": 3.135056854008371e-07, + "logits/chosen": -16.152729034423828, + "logits/rejected": -16.002233505249023, + "logps/chosen": -1.8163830041885376, + "logps/rejected": -2.228738784790039, + "loss": 2.8591, + "rewards/accuracies": 0.75, + "rewards/chosen": -18.163829803466797, + "rewards/margins": 4.123559474945068, + "rewards/rejected": -22.28738784790039, + "step": 285 + }, + { + "epoch": 0.6229860365198711, + "grad_norm": 113.24003069429473, + "learning_rate": 2.988942893956833e-07, + "logits/chosen": -15.338768005371094, + "logits/rejected": -15.249606132507324, + "logps/chosen": -1.8918412923812866, + "logps/rejected": -2.32609224319458, + "loss": 3.0229, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -18.918415069580078, + "rewards/margins": 4.342508792877197, + "rewards/rejected": -23.260921478271484, + "step": 290 + }, + { + "epoch": 0.6337271750805585, + "grad_norm": 151.8876206935079, + "learning_rate": 2.844256551091911e-07, + "logits/chosen": -16.8232421875, + "logits/rejected": -16.842761993408203, + "logps/chosen": -1.9518108367919922, + "logps/rejected": -2.4939913749694824, + "loss": 2.833, + "rewards/accuracies": 0.8125, + "rewards/chosen": -19.518108367919922, + "rewards/margins": 5.421802997589111, + "rewards/rejected": -24.939910888671875, + "step": 295 + }, + { + "epoch": 0.644468313641246, + "grad_norm": 273.6603281637076, + "learning_rate": 2.7012021231812664e-07, + "logits/chosen": -16.766956329345703, + "logits/rejected": -16.424999237060547, + "logps/chosen": -1.969435453414917, + "logps/rejected": -2.308202028274536, + "loss": 3.2442, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -19.694355010986328, + "rewards/margins": 3.387664318084717, + "rewards/rejected": -23.082019805908203, + "step": 300 + }, + { + "epoch": 0.6552094522019334, + "grad_norm": 187.01720889425104, + "learning_rate": 2.5599816037212954e-07, + "logits/chosen": -14.743069648742676, + "logits/rejected": -14.623723983764648, + "logps/chosen": -1.8649146556854248, + "logps/rejected": -2.4030163288116455, + "loss": 2.9136, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": -18.64914894104004, + "rewards/margins": 5.381015777587891, + "rewards/rejected": -24.03016471862793, + "step": 305 + }, + { + "epoch": 0.6659505907626209, + "grad_norm": 142.83830462137144, + "learning_rate": 2.4207943967214064e-07, + "logits/chosen": -16.09463119506836, + "logits/rejected": -15.806689262390137, + "logps/chosen": -2.033447742462158, + "logps/rejected": -2.524364471435547, + "loss": 3.1207, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -20.334476470947266, + "rewards/margins": 4.909164905548096, + "rewards/rejected": -25.24364471435547, + "step": 310 + }, + { + "epoch": 0.6766917293233082, + "grad_norm": 106.6308691579763, + "learning_rate": 2.2838370351446547e-07, + "logits/chosen": -15.870585441589355, + "logits/rejected": -15.55876350402832, + "logps/chosen": -1.8023881912231445, + "logps/rejected": -2.277968168258667, + "loss": 2.7641, + "rewards/accuracies": 0.6875, + "rewards/chosen": -18.023881912231445, + "rewards/margins": 4.755801200866699, + "rewards/rejected": -22.779682159423828, + "step": 315 + }, + { + "epoch": 0.6874328678839957, + "grad_norm": 108.7814729394269, + "learning_rate": 2.1493029034023188e-07, + "logits/chosen": -15.210580825805664, + "logits/rejected": -15.057415962219238, + "logps/chosen": -1.789072036743164, + "logps/rejected": -2.2775633335113525, + "loss": 2.8863, + "rewards/accuracies": 0.75, + "rewards/chosen": -17.89072036743164, + "rewards/margins": 4.884912014007568, + "rewards/rejected": -22.775630950927734, + "step": 320 + }, + { + "epoch": 0.6981740064446831, + "grad_norm": 180.86259023584043, + "learning_rate": 2.0173819642942376e-07, + "logits/chosen": -14.378689765930176, + "logits/rejected": -14.201916694641113, + "logps/chosen": -1.9404065608978271, + "logps/rejected": -2.5850563049316406, + "loss": 2.9848, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -19.404064178466797, + "rewards/margins": 6.446499824523926, + "rewards/rejected": -25.85056495666504, + "step": 325 + }, + { + "epoch": 0.7089151450053706, + "grad_norm": 132.13798361660807, + "learning_rate": 1.888260490780485e-07, + "logits/chosen": -14.281087875366211, + "logits/rejected": -14.154438972473145, + "logps/chosen": -1.7613548040390015, + "logps/rejected": -2.238495349884033, + "loss": 3.0495, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -17.61355209350586, + "rewards/margins": 4.771404266357422, + "rewards/rejected": -22.384952545166016, + "step": 330 + }, + { + "epoch": 0.719656283566058, + "grad_norm": 127.44733826541191, + "learning_rate": 1.7621208029631078e-07, + "logits/chosen": -14.766406059265137, + "logits/rejected": -14.667470932006836, + "logps/chosen": -1.9043552875518799, + "logps/rejected": -2.4970054626464844, + "loss": 2.9109, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -19.04355239868164, + "rewards/margins": 5.9265007972717285, + "rewards/rejected": -24.970054626464844, + "step": 335 + }, + { + "epoch": 0.7303974221267454, + "grad_norm": 114.16195914051926, + "learning_rate": 1.6391410106493227e-07, + "logits/chosen": -14.881872177124023, + "logits/rejected": -14.627456665039062, + "logps/chosen": -1.9786325693130493, + "logps/rejected": -2.5424978733062744, + "loss": 2.8817, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -19.786325454711914, + "rewards/margins": 5.63865327835083, + "rewards/rejected": -25.424976348876953, + "step": 340 + }, + { + "epoch": 0.7411385606874329, + "grad_norm": 99.76652109445614, + "learning_rate": 1.5194947618596673e-07, + "logits/chosen": -15.026782035827637, + "logits/rejected": -14.58587646484375, + "logps/chosen": -1.9211170673370361, + "logps/rejected": -2.394774913787842, + "loss": 3.0269, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -19.211170196533203, + "rewards/margins": 4.736577033996582, + "rewards/rejected": -23.9477481842041, + "step": 345 + }, + { + "epoch": 0.7518796992481203, + "grad_norm": 107.2380444923777, + "learning_rate": 1.4033509976362083e-07, + "logits/chosen": -15.670697212219238, + "logits/rejected": -15.579752922058105, + "logps/chosen": -1.9338979721069336, + "logps/rejected": -2.3520257472991943, + "loss": 2.9892, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -19.338979721069336, + "rewards/margins": 4.181277751922607, + "rewards/rejected": -23.52025604248047, + "step": 350 + }, + { + "epoch": 0.7626208378088077, + "grad_norm": 108.07049767560719, + "learning_rate": 1.2908737134970363e-07, + "logits/chosen": -14.5513334274292, + "logits/rejected": -14.516873359680176, + "logps/chosen": -1.8658726215362549, + "logps/rejected": -2.476576566696167, + "loss": 3.1196, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -18.65872573852539, + "rewards/margins": 6.1070404052734375, + "rewards/rejected": -24.765766143798828, + "step": 355 + }, + { + "epoch": 0.7733619763694952, + "grad_norm": 103.96338647941381, + "learning_rate": 1.1822217278738515e-07, + "logits/chosen": -15.559527397155762, + "logits/rejected": -15.545167922973633, + "logps/chosen": -1.9091074466705322, + "logps/rejected": -2.4284536838531494, + "loss": 3.0461, + "rewards/accuracies": 0.75, + "rewards/chosen": -19.091075897216797, + "rewards/margins": 5.193462371826172, + "rewards/rejected": -24.284536361694336, + "step": 360 + }, + { + "epoch": 0.7841031149301826, + "grad_norm": 138.78303459839336, + "learning_rate": 1.0775484578596241e-07, + "logits/chosen": -15.669352531433105, + "logits/rejected": -15.537897109985352, + "logps/chosen": -1.94468092918396, + "logps/rejected": -2.5436809062957764, + "loss": 2.7974, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -19.44681167602539, + "rewards/margins": 5.989997386932373, + "rewards/rejected": -25.436809539794922, + "step": 365 + }, + { + "epoch": 0.7948442534908701, + "grad_norm": 122.14734643501443, + "learning_rate": 9.770017025829673e-08, + "logits/chosen": -15.961019515991211, + "logits/rejected": -15.95417308807373, + "logps/chosen": -2.1812241077423096, + "logps/rejected": -2.6908135414123535, + "loss": 2.6374, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -21.812240600585938, + "rewards/margins": 5.095890998840332, + "rewards/rejected": -26.908132553100586, + "step": 370 + }, + { + "epoch": 0.8055853920515574, + "grad_norm": 114.19485857094502, + "learning_rate": 8.807234345151027e-08, + "logits/chosen": -14.920249938964844, + "logits/rejected": -14.890344619750977, + "logps/chosen": -2.0413661003112793, + "logps/rejected": -2.6543760299682617, + "loss": 2.8891, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": -20.41366195678711, + "rewards/margins": 6.13009786605835, + "rewards/rejected": -26.543758392333984, + "step": 375 + }, + { + "epoch": 0.8163265306122449, + "grad_norm": 184.4289644042616, + "learning_rate": 7.888495990040924e-08, + "logits/chosen": -13.656982421875, + "logits/rejected": -13.701431274414062, + "logps/chosen": -2.011772871017456, + "logps/rejected": -2.7216084003448486, + "loss": 2.9801, + "rewards/accuracies": 0.8374999761581421, + "rewards/chosen": -20.117727279663086, + "rewards/margins": 7.098354339599609, + "rewards/rejected": -27.21608543395996, + "step": 380 + }, + { + "epoch": 0.8270676691729323, + "grad_norm": 132.46593111411627, + "learning_rate": 7.015099223193943e-08, + "logits/chosen": -15.658330917358398, + "logits/rejected": -15.693890571594238, + "logps/chosen": -1.9581212997436523, + "logps/rejected": -2.5001060962677, + "loss": 2.8611, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -19.58121681213379, + "rewards/margins": 5.419846534729004, + "rewards/rejected": -25.001062393188477, + "step": 385 + }, + { + "epoch": 0.8378088077336198, + "grad_norm": 109.73562851103766, + "learning_rate": 6.188277284777857e-08, + "logits/chosen": -14.48884391784668, + "logits/rejected": -13.800481796264648, + "logps/chosen": -1.9877732992172241, + "logps/rejected": -2.5746169090270996, + "loss": 2.8295, + "rewards/accuracies": 0.8374999761581421, + "rewards/chosen": -19.877731323242188, + "rewards/margins": 5.868436813354492, + "rewards/rejected": -25.746166229248047, + "step": 390 + }, + { + "epoch": 0.8485499462943072, + "grad_norm": 119.04235166242496, + "learning_rate": 5.409197651092965e-08, + "logits/chosen": -15.729510307312012, + "logits/rejected": -15.620780944824219, + "logps/chosen": -2.1660141944885254, + "logps/rejected": -2.667978286743164, + "loss": 2.7298, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -21.660140991210938, + "rewards/margins": 5.019640922546387, + "rewards/rejected": -26.679784774780273, + "step": 395 + }, + { + "epoch": 0.8592910848549946, + "grad_norm": 105.88618147362118, + "learning_rate": 4.678960386090298e-08, + "logits/chosen": -15.191770553588867, + "logits/rejected": -15.158576965332031, + "logps/chosen": -1.9015194177627563, + "logps/rejected": -2.5145716667175293, + "loss": 2.7402, + "rewards/accuracies": 0.8125, + "rewards/chosen": -19.015193939208984, + "rewards/margins": 6.130521297454834, + "rewards/rejected": -25.145715713500977, + "step": 400 + }, + { + "epoch": 0.8700322234156821, + "grad_norm": 114.47737417347113, + "learning_rate": 3.998596588076366e-08, + "logits/chosen": -13.7559814453125, + "logits/rejected": -13.483953475952148, + "logps/chosen": -1.9659799337387085, + "logps/rejected": -2.390488862991333, + "loss": 3.1028, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -19.659801483154297, + "rewards/margins": 4.245090484619141, + "rewards/rejected": -23.904891967773438, + "step": 405 + }, + { + "epoch": 0.8807733619763695, + "grad_norm": 124.82158209754382, + "learning_rate": 3.3690669337976996e-08, + "logits/chosen": -15.061103820800781, + "logits/rejected": -14.891242980957031, + "logps/chosen": -1.8727645874023438, + "logps/rejected": -2.3343753814697266, + "loss": 2.7051, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -18.727645874023438, + "rewards/margins": 4.6161088943481445, + "rewards/rejected": -23.343753814697266, + "step": 410 + }, + { + "epoch": 0.8915145005370569, + "grad_norm": 127.84134494966216, + "learning_rate": 2.7912603219609798e-08, + "logits/chosen": -15.650156021118164, + "logits/rejected": -15.531412124633789, + "logps/chosen": -2.0558724403381348, + "logps/rejected": -2.456268787384033, + "loss": 2.7054, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -20.558725357055664, + "rewards/margins": 4.003961563110352, + "rewards/rejected": -24.562685012817383, + "step": 415 + }, + { + "epoch": 0.9022556390977443, + "grad_norm": 121.68417066288369, + "learning_rate": 2.265992618104029e-08, + "logits/chosen": -15.883665084838867, + "logits/rejected": -15.859227180480957, + "logps/chosen": -2.1571857929229736, + "logps/rejected": -2.7032248973846436, + "loss": 2.7706, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -21.571857452392578, + "rewards/margins": 5.460390090942383, + "rewards/rejected": -27.03224754333496, + "step": 420 + }, + { + "epoch": 0.9129967776584318, + "grad_norm": 205.5451817035243, + "learning_rate": 1.7940055025900304e-08, + "logits/chosen": -14.086555480957031, + "logits/rejected": -13.882139205932617, + "logps/chosen": -2.0481374263763428, + "logps/rejected": -2.428682804107666, + "loss": 3.0577, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -20.481372833251953, + "rewards/margins": 3.8054566383361816, + "rewards/rejected": -24.28683090209961, + "step": 425 + }, + { + "epoch": 0.9237379162191193, + "grad_norm": 119.46095967048043, + "learning_rate": 1.3759654233514817e-08, + "logits/chosen": -14.88987922668457, + "logits/rejected": -14.742823600769043, + "logps/chosen": -1.941057562828064, + "logps/rejected": -2.4432146549224854, + "loss": 2.8075, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -19.41057586669922, + "rewards/margins": 5.021571636199951, + "rewards/rejected": -24.432147979736328, + "step": 430 + }, + { + "epoch": 0.9344790547798066, + "grad_norm": 112.52416047730715, + "learning_rate": 1.0124626548627402e-08, + "logits/chosen": -15.55200481414795, + "logits/rejected": -15.57677936553955, + "logps/chosen": -2.0663094520568848, + "logps/rejected": -2.723595380783081, + "loss": 2.7934, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -20.663097381591797, + "rewards/margins": 6.5728559494018555, + "rewards/rejected": -27.235950469970703, + "step": 435 + }, + { + "epoch": 0.9452201933404941, + "grad_norm": 95.43516446345191, + "learning_rate": 7.040104646698042e-09, + "logits/chosen": -14.149500846862793, + "logits/rejected": -14.121160507202148, + "logps/chosen": -2.1648197174072266, + "logps/rejected": -2.803515911102295, + "loss": 2.7258, + "rewards/accuracies": 0.75, + "rewards/chosen": -21.648197174072266, + "rewards/margins": 6.386962890625, + "rewards/rejected": -28.035160064697266, + "step": 440 + }, + { + "epoch": 0.9559613319011815, + "grad_norm": 105.32519506523218, + "learning_rate": 4.510443886542114e-09, + "logits/chosen": -15.509679794311523, + "logits/rejected": -15.587133407592773, + "logps/chosen": -2.019160509109497, + "logps/rejected": -2.5319721698760986, + "loss": 2.8015, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -20.191600799560547, + "rewards/margins": 5.128118991851807, + "rewards/rejected": -25.319721221923828, + "step": 445 + }, + { + "epoch": 0.966702470461869, + "grad_norm": 142.71634790119586, + "learning_rate": 2.539216160544333e-09, + "logits/chosen": -15.480878829956055, + "logits/rejected": -15.102048873901367, + "logps/chosen": -2.1352379322052, + "logps/rejected": -2.589895486831665, + "loss": 2.889, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -21.35237693786621, + "rewards/margins": 4.546576023101807, + "rewards/rejected": -25.898956298828125, + "step": 450 + }, + { + "epoch": 0.9774436090225563, + "grad_norm": 123.94313579799326, + "learning_rate": 1.1292048511303054e-09, + "logits/chosen": -14.889852523803711, + "logits/rejected": -15.104809761047363, + "logps/chosen": -1.9709218740463257, + "logps/rejected": -2.4727888107299805, + "loss": 2.9813, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -19.709218978881836, + "rewards/margins": 5.018665313720703, + "rewards/rejected": -24.727886199951172, + "step": 455 + }, + { + "epoch": 0.9881847475832438, + "grad_norm": 100.41632245481235, + "learning_rate": 2.82400900618418e-10, + "logits/chosen": -15.215599060058594, + "logits/rejected": -15.19567584991455, + "logps/chosen": -1.937787652015686, + "logps/rejected": -2.593761920928955, + "loss": 2.5524, + "rewards/accuracies": 0.8125, + "rewards/chosen": -19.377878189086914, + "rewards/margins": 6.559741020202637, + "rewards/rejected": -25.937618255615234, + "step": 460 + }, + { + "epoch": 0.9989258861439313, + "grad_norm": 139.5358860215804, + "learning_rate": 0.0, + "logits/chosen": -15.462666511535645, + "logits/rejected": -15.356382369995117, + "logps/chosen": -2.155303716659546, + "logps/rejected": -2.6153688430786133, + "loss": 2.5621, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -21.55303955078125, + "rewards/margins": 4.600649356842041, + "rewards/rejected": -26.1536865234375, + "step": 465 + }, + { + "epoch": 0.9989258861439313, + "step": 465, + "total_flos": 0.0, + "train_loss": 3.5518602760889197, + "train_runtime": 6148.2375, + "train_samples_per_second": 9.689, + "train_steps_per_second": 0.076 + } + ], + "logging_steps": 5, + "max_steps": 465, + "num_input_tokens_seen": 0, + "num_train_epochs": 1, + "save_steps": 1000000, + "stateful_callbacks": { + "TrainerControl": { + "args": { + "should_epoch_stop": false, + "should_evaluate": false, + "should_log": false, + "should_save": true, + "should_training_stop": true + }, + "attributes": {} + } + }, + "total_flos": 0.0, + "train_batch_size": 2, + "trial_name": null, + "trial_params": null +} diff --git a/training_args.bin b/training_args.bin new file mode 100644 index 0000000..b825936 --- /dev/null +++ b/training_args.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cbb910a253d0edeb0d46fe4d5fd2c5f93a72c20c552645a36b1e11ccda97adbd +size 7096