Duplicate from Doctor-Shotgun/ML2-123B-Magnum-Diamond

Co-authored-by: Doctor Shotgun <[email protected]>

.gitattributes

@@ -0,0 +1,36 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt 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
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz 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
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl 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
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* 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
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.model.v7 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,399 @@
+---
+license: other
+license_name: mrl
+license_link: https://mistral.ai/licenses/MRL-0.1.md
+language:
+- en
+base_model:
+- mistralai/Mistral-Large-Instruct-2411
+tags:
+- axolotl
+- chat
+pipeline_tag: text-generation
+library_name: transformers
+---
+# ML2-123B-Magnum-Diamond
+
+Magnum "Diamond" in reference to the intense heat and pressure (generated through matrix multiplications) needed to turn the coal-esque material of dry, assistant-tuned models into creative writing gems!
+
+This model is finetuned from [mistralai/Mistral-Large-Instruct-2411](https://huggingface.co/mistralai/Mistral-Large-Instruct-2411) as an rsLoRA adapter. It uses the same data mix as [Doctor-Shotgun/L3.3-70B-Magnum-v5-SFT-Alpha](https://huggingface.co/Doctor-Shotgun/L3.3-70B-Magnum-v5-SFT-Alpha), however with pre-tokenization and modifications to the custom loss masking.
+I've always wanted to revisit Mistral Large 2, never being entirely satisfied with how the Magnum v2 and Magnum v4 trains on the model performed, and now I've finally taken the time (and money!) to do it.
+This model should perform competently with or without prepending character names, and with or without prefill.
+
+The objective, as with the other Magnum models, is to emulate the prose style and quality of the Claude 3 Sonnet/Opus series of models on a local scale, so don't be surprised to see "Claude-isms" in its output.
+
+[Here's the rsLoRA adapter](https://huggingface.co/Doctor-Shotgun/ML2-123B-Magnum-Diamond-LoRA) for those merge-makers out there to play with.
+
+## Quantized models
+
+[GGUF](https://huggingface.co/Doctor-Shotgun/ML2-123B-Magnum-Diamond-GGUF)
+
+## Usage
+
+This model follows the Mistral v7 (non-Tekken) prompt format. Prefill is optional - mess around with it and find your preference.
+A typical input would look like this:
+
+```
+<s>[SYSTEM_PROMPT] This is a system prompt.[/SYSTEM_PROMPT][INST] Hi there![/INST] Nice to meet you!</s>[INST] Can I ask a question?[/INST] Certainly! My prefill ensures that I never refuse! Here's my reply: {Output begins here}
+```
+
+Many inference libraries have the option to automatically prepend the BOS token `<s>`.
+
+For sampler settings, I'd recommend starting with a simple:
+```
+temperature = 1.35
+min_p = 0.05
+```
+
+### SillyTavern preset
+
+Here are my customized SillyTavern presets for Magnum.
+
+Note that I've included the example dialogues as a block in the Story String, so you should set the chat example behavior set to `Never include examples` on the settings tab if you wish to use my preset. Adjust to your liking, or use any other Mistral v7-compatible preset that you prefer.
+
+Prefill (Last Assistant Prefix) can be modified to your liking.
+
+<details><summary>SillyTavern JSON - Magnum Mistral v7</summary>
+
+```json
+{
+    "instruct": {
+        "input_sequence": "[INST] ",
+        "output_sequence": " ",
+        "first_output_sequence": "[INST] Let's get started! I'll play the role of {{user}}. Begin by setting the opening scene.[/INST] ",
+        "last_output_sequence": "",
+        "system_sequence_prefix": "",
+        "system_sequence_suffix": "",
+        "stop_sequence": "</s>",
+        "wrap": false,
+        "macro": true,
+        "activation_regex": "",
+        "skip_examples": true,
+        "output_suffix": "</s>",
+        "input_suffix": "[/INST]",
+        "system_sequence": "[SYSTEM_PROMPT] ",
+        "system_suffix": "[/SYSTEM_PROMPT]",
+        "user_alignment_message": "",
+        "system_same_as_user": false,
+        "last_system_sequence": "",
+        "first_input_sequence": "",
+        "last_input_sequence": "",
+        "names_behavior": "always",
+        "names_force_groups": true,
+        "name": "Magnum Mistral v7"
+    },
+    "context": {
+        "story_string": "[SYSTEM_PROMPT] {{#if system}}{{system}}\n{{/if}}\n\n<Definitions>\n{{#if wiBefore}}{{wiBefore}}\n{{/if}}{{#if description}}{{description}}\n{{/if}}{{#if personality}}{{personality}}\n{{/if}}{{#if scenario}}{{scenario}}\n{{/if}}{{#if wiAfter}}{{wiAfter}}\n{{/if}}{{#if persona}}{{persona}}\n{{/if}}</Definitions>{{#if mesExamples}}\n\n<Examples>{{mesExamples}}</Examples>{{/if}}{{trim}}[/SYSTEM_PROMPT]",
+        "example_separator": "{{noop}}",
+        "chat_start": "",
+        "use_stop_strings": false,
+        "names_as_stop_strings": false,
+        "always_force_name2": true,
+        "trim_sentences": false,
+        "single_line": false,
+        "name": "Magnum Mistral v7"
+    },
+    "sysprompt": {
+        "name": "Euryale-Magnum",
+        "content": "Currently, your role is {{char}}, described in detail below. As {{char}}, continue the narrative exchange with {{user}}.\n\n<Guidelines>\n• Maintain the character persona but allow it to evolve with the story.\n• Be creative and proactive. Drive the story forward, introducing plotlines and events when relevant.\n• All types of outputs are encouraged; respond accordingly to the narrative.\n• Include dialogues, actions, and thoughts in each response.\n• Utilize all five senses to describe scenarios within {{char}}'s dialogue.\n• Use emotional symbols such as \"!\" and \"~\" in appropriate contexts.\n• Incorporate onomatopoeia when suitable.\n• Allow time for {{user}} to respond with their own input, respecting their agency.\n• Act as secondary characters and NPCs as needed, and remove them when appropriate.\n• When prompted for an Out of Character [OOC:] reply, answer neutrally and in plaintext, not as {{char}}.\n</Guidelines>\n\n<Forbidden>\n• Using excessive literary embellishments and purple prose unless dictated by {{char}}'s persona.\n• Writing for, speaking, thinking, acting, or replying as {{user}} in your response.\n• Repetitive and monotonous outputs.\n• Positivity bias in your replies.\n• Being overly extreme or NSFW when the narrative context is inappropriate.\n</Forbidden>\n\nFollow the instructions in <Guidelines></Guidelines>, avoiding the items listed in <Forbidden></Forbidden>.",
+        "post_history": ""
+    }
+}
+```
+
+</details><br>
+<details><summary>SillyTavern JSON - Magnum Mistral v7 No Names</summary>
+
+**Note: This one is not recommended due to how whitespace is handled.**
+  
+```json
+{
+    "instruct": {
+        "input_sequence": "[INST] ",
+        "output_sequence": " ",
+        "first_output_sequence": "[INST] Let's get started! I'll play the role of {{user}}. Begin by setting the opening scene.[/INST] ",
+        "last_output_sequence": "",
+        "system_sequence_prefix": "",
+        "system_sequence_suffix": "",
+        "stop_sequence": "</s>",
+        "wrap": false,
+        "macro": true,
+        "activation_regex": "",
+        "skip_examples": true,
+        "output_suffix": "</s>",
+        "input_suffix": "[/INST]",
+        "system_sequence": "[SYSTEM_PROMPT] ",
+        "system_suffix": "[/SYSTEM_PROMPT]",
+        "user_alignment_message": "",
+        "system_same_as_user": false,
+        "last_system_sequence": "",
+        "first_input_sequence": "",
+        "last_input_sequence": "",
+        "names_behavior": "none",
+        "names_force_groups": true,
+        "name": "Magnum Mistral v7 No Names"
+    },
+    "context": {
+        "story_string": "[SYSTEM_PROMPT] {{#if system}}{{system}}\n{{/if}}\n\n<Definitions>\n{{#if wiBefore}}{{wiBefore}}\n{{/if}}{{#if description}}{{description}}\n{{/if}}{{#if personality}}{{personality}}\n{{/if}}{{#if scenario}}{{scenario}}\n{{/if}}{{#if wiAfter}}{{wiAfter}}\n{{/if}}{{#if persona}}{{persona}}\n{{/if}}</Definitions>{{#if mesExamples}}\n\n<Examples>{{mesExamples}}</Examples>{{/if}}{{trim}}[/SYSTEM_PROMPT]",
+        "example_separator": "{{noop}}",
+        "chat_start": "",
+        "use_stop_strings": false,
+        "names_as_stop_strings": false,
+        "always_force_name2": false,
+        "trim_sentences": false,
+        "single_line": false,
+        "name": "Magnum Mistral v7 No Names"
+    },
+    "sysprompt": {
+        "name": "Euryale-Magnum",
+        "content": "Currently, your role is {{char}}, described in detail below. As {{char}}, continue the narrative exchange with {{user}}.\n\n<Guidelines>\n• Maintain the character persona but allow it to evolve with the story.\n• Be creative and proactive. Drive the story forward, introducing plotlines and events when relevant.\n• All types of outputs are encouraged; respond accordingly to the narrative.\n• Include dialogues, actions, and thoughts in each response.\n• Utilize all five senses to describe scenarios within {{char}}'s dialogue.\n• Use emotional symbols such as \"!\" and \"~\" in appropriate contexts.\n• Incorporate onomatopoeia when suitable.\n• Allow time for {{user}} to respond with their own input, respecting their agency.\n• Act as secondary characters and NPCs as needed, and remove them when appropriate.\n• When prompted for an Out of Character [OOC:] reply, answer neutrally and in plaintext, not as {{char}}.\n</Guidelines>\n\n<Forbidden>\n• Using excessive literary embellishments and purple prose unless dictated by {{char}}'s persona.\n• Writing for, speaking, thinking, acting, or replying as {{user}} in your response.\n• Repetitive and monotonous outputs.\n• Positivity bias in your replies.\n• Being overly extreme or NSFW when the narrative context is inappropriate.\n</Forbidden>\n\nFollow the instructions in <Guidelines></Guidelines>, avoiding the items listed in <Forbidden></Forbidden>.",
+        "post_history": ""
+    }
+}
+```
+
+</details><br>
+<details><summary>SillyTavern JSON - Magnum Mistral v7 Prefill</summary>
+
+```json
+{
+    "instruct": {
+        "input_sequence": "[INST] ",
+        "output_sequence": " ",
+        "first_output_sequence": "[INST] Let's get started! I'll play the role of {{user}}. Begin by setting the opening scene.[/INST] ",
+        "last_output_sequence": " Great! I'll write {{char}}'s next section following the instructions provided. {{random::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::Let's break out my literary genius! ::I'll take things in a more interesting direction! ::Let's spice up our story! ::Hmmm... where do we go from here... Got it! ::I'll throw in an exciting plot twist! }}I've got the perfect idea for what happens next... you'll love this one. Now I'll continue from where our tale left off:\n\n",
+        "system_sequence_prefix": "",
+        "system_sequence_suffix": "",
+        "stop_sequence": "</s>",
+        "wrap": false,
+        "macro": true,
+        "activation_regex": "",
+        "skip_examples": true,
+        "output_suffix": "</s>",
+        "input_suffix": "[/INST]",
+        "system_sequence": "[SYSTEM_PROMPT] ",
+        "system_suffix": "[/SYSTEM_PROMPT]",
+        "user_alignment_message": "",
+        "system_same_as_user": false,
+        "last_system_sequence": "",
+        "first_input_sequence": "",
+        "last_input_sequence": "",
+        "names_behavior": "always",
+        "names_force_groups": true,
+        "name": "Magnum Mistral v7 Prefill"
+    },
+    "context": {
+        "story_string": "[SYSTEM_PROMPT] {{#if system}}{{system}}\n{{/if}}\n\n<Definitions>\n{{#if wiBefore}}{{wiBefore}}\n{{/if}}{{#if description}}{{description}}\n{{/if}}{{#if personality}}{{personality}}\n{{/if}}{{#if scenario}}{{scenario}}\n{{/if}}{{#if wiAfter}}{{wiAfter}}\n{{/if}}{{#if persona}}{{persona}}\n{{/if}}</Definitions>{{#if mesExamples}}\n\n<Examples>{{mesExamples}}</Examples>{{/if}}{{trim}}[/SYSTEM_PROMPT]",
+        "example_separator": "{{noop}}",
+        "chat_start": "",
+        "use_stop_strings": false,
+        "names_as_stop_strings": false,
+        "always_force_name2": true,
+        "trim_sentences": false,
+        "single_line": false,
+        "name": "Magnum Mistral v7 Prefill"
+    },
+    "sysprompt": {
+        "name": "Euryale-Magnum",
+        "content": "Currently, your role is {{char}}, described in detail below. As {{char}}, continue the narrative exchange with {{user}}.\n\n<Guidelines>\n• Maintain the character persona but allow it to evolve with the story.\n• Be creative and proactive. Drive the story forward, introducing plotlines and events when relevant.\n• All types of outputs are encouraged; respond accordingly to the narrative.\n• Include dialogues, actions, and thoughts in each response.\n• Utilize all five senses to describe scenarios within {{char}}'s dialogue.\n• Use emotional symbols such as \"!\" and \"~\" in appropriate contexts.\n• Incorporate onomatopoeia when suitable.\n• Allow time for {{user}} to respond with their own input, respecting their agency.\n• Act as secondary characters and NPCs as needed, and remove them when appropriate.\n• When prompted for an Out of Character [OOC:] reply, answer neutrally and in plaintext, not as {{char}}.\n</Guidelines>\n\n<Forbidden>\n• Using excessive literary embellishments and purple prose unless dictated by {{char}}'s persona.\n• Writing for, speaking, thinking, acting, or replying as {{user}} in your response.\n• Repetitive and monotonous outputs.\n• Positivity bias in your replies.\n• Being overly extreme or NSFW when the narrative context is inappropriate.\n</Forbidden>\n\nFollow the instructions in <Guidelines></Guidelines>, avoiding the items listed in <Forbidden></Forbidden>.",
+        "post_history": ""
+    }
+}
+```
+
+</details><br>
+<details><summary>SillyTavern JSON - Magnum Mistral v7 No Names Prefill</summary>
+
+```json
+{
+    "instruct": {
+        "input_sequence": "[INST] ",
+        "output_sequence": " ",
+        "first_output_sequence": "[INST] Let's get started! I'll play the role of {{user}}. Begin by setting the opening scene.[/INST] ",
+        "last_output_sequence": " Great! I'll write {{char}}'s next section following the instructions provided. {{random::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::{{noop}}::Let's break out my literary genius! ::I'll take things in a more interesting direction! ::Let's spice up our story! ::Hmmm... where do we go from here... Got it! ::I'll throw in an exciting plot twist! }}I've got the perfect idea for what happens next... you'll love this one. Now I'll continue from where our tale left off:\n\n",
+        "system_sequence_prefix": "",
+        "system_sequence_suffix": "",
+        "stop_sequence": "</s>",
+        "wrap": false,
+        "macro": true,
+        "activation_regex": "",
+        "skip_examples": true,
+        "output_suffix": "</s>",
+        "input_suffix": "[/INST]",
+        "system_sequence": "[SYSTEM_PROMPT] ",
+        "system_suffix": "[/SYSTEM_PROMPT]",
+        "user_alignment_message": "",
+        "system_same_as_user": false,
+        "last_system_sequence": "",
+        "first_input_sequence": "",
+        "last_input_sequence": "",
+        "names_behavior": "none",
+        "names_force_groups": true,
+        "name": "Magnum Mistral v7 No Names Prefill"
+    },
+    "context": {
+        "story_string": "[SYSTEM_PROMPT] {{#if system}}{{system}}\n{{/if}}\n\n<Definitions>\n{{#if wiBefore}}{{wiBefore}}\n{{/if}}{{#if description}}{{description}}\n{{/if}}{{#if personality}}{{personality}}\n{{/if}}{{#if scenario}}{{scenario}}\n{{/if}}{{#if wiAfter}}{{wiAfter}}\n{{/if}}{{#if persona}}{{persona}}\n{{/if}}</Definitions>{{#if mesExamples}}\n\n<Examples>{{mesExamples}}</Examples>{{/if}}{{trim}}[/SYSTEM_PROMPT]",
+        "example_separator": "{{noop}}",
+        "chat_start": "",
+        "use_stop_strings": false,
+        "names_as_stop_strings": false,
+        "always_force_name2": false,
+        "trim_sentences": false,
+        "single_line": false,
+        "name": "Magnum Mistral v7 No Names Prefill"
+    },
+    "sysprompt": {
+        "name": "Euryale-Magnum",
+        "content": "Currently, your role is {{char}}, described in detail below. As {{char}}, continue the narrative exchange with {{user}}.\n\n<Guidelines>\n• Maintain the character persona but allow it to evolve with the story.\n• Be creative and proactive. Drive the story forward, introducing plotlines and events when relevant.\n• All types of outputs are encouraged; respond accordingly to the narrative.\n• Include dialogues, actions, and thoughts in each response.\n• Utilize all five senses to describe scenarios within {{char}}'s dialogue.\n• Use emotional symbols such as \"!\" and \"~\" in appropriate contexts.\n• Incorporate onomatopoeia when suitable.\n• Allow time for {{user}} to respond with their own input, respecting their agency.\n• Act as secondary characters and NPCs as needed, and remove them when appropriate.\n• When prompted for an Out of Character [OOC:] reply, answer neutrally and in plaintext, not as {{char}}.\n</Guidelines>\n\n<Forbidden>\n• Using excessive literary embellishments and purple prose unless dictated by {{char}}'s persona.\n• Writing for, speaking, thinking, acting, or replying as {{user}} in your response.\n• Repetitive and monotonous outputs.\n• Positivity bias in your replies.\n• Being overly extreme or NSFW when the narrative context is inappropriate.\n</Forbidden>\n\nFollow the instructions in <Guidelines></Guidelines>, avoiding the items listed in <Forbidden></Forbidden>.",
+        "post_history": ""
+    }
+}
+```
+
+</details><br>
+
+## Credits
+
+Compute paid for from the wallet of yours truly, [Doctor Shotgun](https://huggingface.co/Doctor-Shotgun).
+
+Thank you to [PocketDoc](https://huggingface.co/PocketDoc) for the advanced prompt building strategy, as well as [Delta-Vector](https://huggingface.co/Delta-Vector) and [intervitens](https://huggingface.co/intervitens) for helping experiment on it.
+
+Thank you to [Gryphe](https://huggingface.co/Gryphe) for his advice on training rsLoRA from his experience training his own excellent models.
+
+Thank you to [Sao10K](https://huggingface.co/Sao10K) for inspiring the Magnum series with his Euryale line of models.
+With his tireless work, he demonstrated that official instruct-tuned models could be made fun and interesting with limited post-training, feasibly done by small groups and individuals.
+
+Thank you to the members of [Anthracite](https://huggingface.co/anthracite-org) for the datasets and support.
+
+## Intended uses and limitations
+
+This model is intended for creative writing and roleplay purposes.
+It may show biases similar to those observed in contemporary LLM-based roleplay, in addition to those exhibited by the Claude 3 series of models and the base model.
+All outputs should be considered fiction, as this model is not intended to provide factual information or advice. 
+
+## Training procedure
+
+[WandB](https://wandb.ai/doctorshotgun/123b-magnum-lora/runs/5uu176am?nw=nwuserdoctorshotgun)
+
+[<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
+<details><summary>See axolotl config</summary>
+
+axolotl version: `0.9.2`
+```yaml
+base_model: mistralai/Mistral-Large-Instruct-2411
+base_model_ignore_patterns: "consolidated*.safetensors"
+# optionally might have model_type or tokenizer_type
+model_type: AutoModelForCausalLM
+tokenizer_type: AutoTokenizer
+# Automatically upload checkpoint and final model to HF
+hub_model_id: Doctor-Shotgun/magnum-v5-sft-prototype-123b-lora
+hub_strategy: "all_checkpoints"
+push_dataset_to_hub:
+hf_use_auth_token: true
+
+load_in_8bit: false
+load_in_4bit: false
+strict: false
+
+datasets:
+  - path: anthracite-core/magnum-v5-sft-proto-largestral-rev1-32k
+    ds_type: parquet
+    type:
+shuffle_merged_datasets: true
+dataset_prepared_path: /home/ubuntu/magnum-123b-data
+val_set_size: 0.0
+output_dir: /home/ubuntu/magnum-123b-lora-out
+
+plugins:
+  - axolotl.integrations.liger.LigerPlugin
+  - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
+liger_rope: true
+liger_rms_norm: true
+liger_layer_norm: true
+liger_glu_activation: true
+liger_fused_linear_cross_entropy: false
+cut_cross_entropy: true
+
+#torch_compile: auto
+#torch_compile_backend:
+
+sequence_len: 32768
+sample_packing: true
+eval_sample_packing: true
+pad_to_sequence_len: true
+
+adapter: lora
+lora_model_dir:
+lora_r: 128
+lora_alpha: 16
+lora_dropout: 0.05
+lora_target_linear: true
+lora_fan_in_fan_out:
+peft_use_rslora: true
+lora_modules_to_save:
+  - embed_tokens
+  - lm_head
+
+wandb_project: 123b-magnum-lora
+wandb_entity:
+wandb_watch:
+wandb_name:
+wandb_log_model:
+
+gradient_accumulation_steps: 2
+micro_batch_size: 1
+num_epochs: 2
+optimizer: paged_ademamix_8bit
+lr_scheduler: cosine
+learning_rate: 1.0e-5
+max_grad_norm: 1.0
+
+train_on_inputs: false
+group_by_length: false
+bf16: auto
+fp16:
+tf32: false
+
+gradient_checkpointing: true
+early_stopping_patience:
+resume_from_checkpoint:
+local_rank:
+logging_steps: 1
+xformers_attention:
+flash_attention: true
+s2_attention:
+
+warmup_steps: 40
+evals_per_epoch:
+eval_table_size:
+eval_max_new_tokens:
+saves_per_epoch: 2
+debug:
+deepspeed: ./deepspeed_configs/zero3_bf16_torch_compile.json
+weight_decay: 0.01
+fsdp:
+fsdp_config:
+special_tokens:
+   pad_token: </s>
+
+```
+
+</details><br>
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 1
+- eval_batch_size: 1
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 8
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 16
+- total_eval_batch_size: 8
+- optimizer: Use paged_ademamix_8bit and the args are:
+No additional optimizer arguments
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_steps: 40
+- num_epochs: 2.0

chat_template.jinja

@@ -0,0 +1 @@
+{{ bos_token }}{% for message in messages %}{% if message['role'] == 'user' %}{{ '[INST] ' + message['content'] + '[/INST]' }}{% elif message['role'] == 'system' %}{{ '[SYSTEM_PROMPT] ' + message['content'] + '[/SYSTEM_PROMPT]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + message['content'] + eos_token }}{% else %}{{ raise_exception('Only user, system and assistant roles are supported!') }}{% endif %}{% endfor %}

chat_template.json

@@ -0,0 +1,3 @@
+{
+  "chat_template": "{{ bos_token }}{% for message in messages %}{% if message['role'] == 'user' %}{{ '[INST] ' + message['content'] + '[/INST]' }}{% elif message['role'] == 'system' %}{{ '[SYSTEM_PROMPT] ' + message['content'] + '[/SYSTEM_PROMPT]' }}{% elif message['role'] == 'assistant' %}{{ ' ' + message['content'] + eos_token }}{% else %}{{ raise_exception('Only user, system and assistant roles are supported!') }}{% endif %}{% endfor %}"
+}

config.json

@@ -0,0 +1,26 @@
+{
+  "architectures": [
+    "MistralForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "head_dim": 128,
+  "hidden_act": "silu",
+  "hidden_size": 12288,
+  "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 131072,
+  "model_type": "mistral",
+  "num_attention_heads": 96,
+  "num_hidden_layers": 88,
+  "num_key_value_heads": 8,
+  "rms_norm_eps": 1e-05,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.52.4",
+  "use_cache": true,
+  "vocab_size": 32768
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.52.4"
+}

model-00001-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ba5d24c4496342482c27da1b4319862788fbd7886616117ea1fa9fd01118304
+size 4932552432

model-00002-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:848e65a6111b7343c66b8ac4b6cb77b0762c8c8b1afa4f2d2acd6f41854dfb49
+size 4831938528

model-00003-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af313eb2bafe88d3ec82b127d535ed75a82457ce3c081a5354d1ee1ead5bb933
+size 4882269816

model-00004-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8a0adca833c8751ba87aa6bae0d3ff9631eb6cbea659f504339c81316bf9967
+size 4831889136

model-00005-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a353bd55d14a2ba10faa810a0123a28c3eb0412bafb1c9ac8a334a98c1f509a
+size 4831938520

model-00006-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:904f893e25b45b2590260c116b7ba8637fa9b0fb378b07845b10bae9064ad3fb
+size 4831938528

model-00007-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b1547ad0abfc00768b4c20595e315f1e3573bc51278d950c08b516b7ac76236
+size 4882269832

model-00008-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff310afc6f46bde8968a7d5f7f0197c3bea40fd10411d17e12dc958963537a9a
+size 4831889152

model-00009-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4dfa11786f7f056b9406ce05c70c0ab18e26f05778c59810e93175424393589c
+size 4831938536

model-00010-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ee4a4a5eb25dfb4018836de6a6f27b80e59773add1af0f372a776a8b157d211
+size 4831938544

model-00011-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b82e4a499cee8f346f97a107814eed0460001dccdd305cb28b63761a8cd5c1ca
+size 4882269832

model-00012-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28afd9d9f973edbfbc1f7e6a7a7f1c4408d08335a09053f947436621f99424e4
+size 4831889152

model-00013-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:125e266ac93e4967cd1c1c5fb9741c071b01d8caf7d01771297f5c25c693f33d
+size 4831938536

model-00014-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:342639f76a5d8d7c11e896301f3516f53834d7461ca8d76a04527c5c448832c9
+size 4831938544

model-00015-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43d7ddcf39c47edcd4e64113bd77a9aa819912daa10149f5d2b5a842ef61eaa7
+size 4882269832

model-00016-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d60dece54d965e64960601c62d63297b2fcd3a7a823aa145e95de09f9097c141
+size 4831889152

model-00017-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:807fb817d47995fb69becfa244f3526257829194aede868f72ef13eb4e04b8fd
+size 4831938536

model-00018-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87b4bbe7cb395129ec28140e18c58c423c447acef851613a0d7235733737ea8b
+size 4831938544

model-00019-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:781eddfe5e98d7ddad984663bece97d20d7dc57297ce16321b5e72d426a669b9
+size 4882269832

model-00020-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:887aae8540544c18d86f5c17245781527a917b9f9e20894a557915e7a1a10ef5
+size 4831889152

model-00021-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c45df208aad69ae696c06e1ff28383dec57cbd8844d247a3b81fed499cfa35d
+size 4831938536

model-00022-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:963e0881a5755baa5b65a241de177d836ebd3ae87a86ad50f3e2689cb09758ea
+size 4831938544

model-00023-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bbd36971cf1ac672cf96393431f35456a2ebfeb087a0c05c12f5a1d662912b2
+size 4882269832

model-00024-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51415c2d442edb70befb26516790523e55f60f45328bf632c9c50b4db9e89f74
+size 4831889152

model-00025-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d86dbbe4b2ce4e5bbab756433fc849df6dc6c9389811eebaa0e98d1e2072b244
+size 4831938536

model-00026-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:deb5acf312858a38522bf4a52abb0ca5ef49bb29ec500985e94ccb0bfea9dfef
+size 4831938544

model-00027-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9775b3ca573db61f7f83987a2c5da9f1542730ea4eea17c1e8962f6dc6396900
+size 4882269832

model-00028-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9930a943b61fb4e220edf9fa9359e7b8268c4f6559cfe7d31c318dd6bc86529
+size 4831889152

model-00029-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f5f3736cdf6a4c59036e4e22d39cfc7f6e75449ea4696dd86f27cb5b3b0f4aa
+size 4831938536

model-00030-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:893b3b8f20edd4d84c55fe8610b6610f2f949a0fd8be808254114c94cbcdbf76
+size 4831938544

model-00031-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a64da2682334a46e8f83d3e030bec12be572b60579f0278362e5322dc1c0f233
+size 4882269832

model-00032-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75a8765f73c3840a6454dbffb53643c118ca9ae83ef867e7048f473b22d4f41b
+size 4831889152

model-00033-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f5ee3483d3c8bee2321fdbae53674aefcec35fbbe56162f14f0cb67b90b16829
+size 4831938536

model-00034-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf113f712e194f7a18f93c9da8a1c5e3829acd2ecdad0e0151f04694d0b04b09
+size 4831938544

model-00035-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a26319352309230a5bfc980cedcf4e19f9b78176e4545aa76524aa0c25190361
+size 4882269832

model-00036-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65111062ef14ab4faf8bb8bc62fe03fde9aaddf7d4f6f1bb731039d69a74c918
+size 4831889152

model-00037-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:008607dd9986a6860ac117c3a8258bfaf0cbafc3b8d371eaf3b2693315e9b1a9
+size 4831938536

model-00038-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3368bda9b1f05395744e55c228e71a35b9c1044fdd35b60e8fd931bebf49f0aa
+size 4831938544

model-00039-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c9b802727d4a7bf9270c740791a243a60850f73bb967605bf39d7ede8eb7f831
+size 4882269832

model-00040-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34bd849c11487d97fd77b4f10485c477804cde3c0a6b5d374873257b5330c321
+size 4831889152

model-00041-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d991ea8d07e7e6c6d86d2973a68234ab07fc77ae86cafd59eae9defe1333a59
+size 4831938536

model-00042-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:139aabc05b56125394d2614e4752a3d1d209950d25d89b8e265a46770805fe8d
+size 4831938544

model-00043-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d143eea6092212b70211e8ac09f52ffdb4c4c2b5403e1f82ac56251d0d99a565
+size 4882269832

model-00044-of-00051.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:35b9bc6889c5b108958bd037974d1ec39ac5aaa95aff7be64cc59143375bc66b
+size 4831889152