Model save

README.md

@@ -0,0 +1,58 @@
+---
+base_model: mistralai/Mixtral-8x7B-Instruct-v0.1
+library_name: transformers
+model_name: mixtral-reasoning-output
+tags:
+- generated_from_trainer
+- trl
+- sft
+licence: license
+---
+
+# Model Card for mixtral-reasoning-output
+
+This model is a fine-tuned version of [mistralai/Mixtral-8x7B-Instruct-v0.1](https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1).
+It has been trained using [TRL](https://github.com/huggingface/trl).
+
+## Quick start
+
+```python
+from transformers import pipeline
+
+question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
+generator = pipeline("text-generation", model="ykarout/mixtral-reasoning-output", device="cuda")
+output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
+print(output["generated_text"])
+```
+
+## Training procedure
+
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/ykar-deloitte/mixtral-reasoning/runs/0gs3k744) 
+
+
+This model was trained with SFT.
+
+### Framework versions
+
+- TRL: 0.18.1
+- Transformers: 4.52.4
+- Pytorch: 2.6.0+cu126
+- Datasets: 3.6.0
+- Tokenizers: 0.21.1
+
+## Citations
+
+
+
+Cite TRL as:
+    
+```bibtex
+@misc{vonwerra2022trl,
+	title        = {{TRL: Transformer Reinforcement Learning}},
+	author       = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
+	year         = 2020,
+	journal      = {GitHub repository},
+	publisher    = {GitHub},
+	howpublished = {\url{https://github.com/huggingface/trl}}
+}
+```

adapter_config.json

@@ -24,13 +24,13 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "w3",
+    "v_proj",
     "k_proj",
+    "o_proj",
     "w1",
-    "v_proj",
-    "w2",
     "q_proj",
-    "o_proj"
+    "w3",
+    "w2"
   ],
   "task_type": "CAUSAL_LM",
   "trainable_token_indices": null,

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:73cc0fc333b4af9cbe787855a761caa8e5e4b95564ce9ce308dc5307ec84135b
+oid sha256:d21a36bce63e6ea32df07db94c49a90776db7ebf4dcf18e078368ab1740bc6df
 size 960760928

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:52bd070477c438c6ef0111d0e4a45f34a42ba8aeac6f9584c965d488939427f5
+oid sha256:545282aed1607b162d59a9796824e9906325bcf235bad06cb7a68a216968e6a2
 size 5752