End of training

README.md

@@ -0,0 +1,186 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: allenai/Olmo-3-1025-7B
+tags:
+- axolotl
+- generated_from_trainer
+datasets:
+- dataset-tfs-mk-IMP-SOS-processed-olmo3-think.jsonl
+model-index:
+- name: O37BB
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+[<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.13.0.dev0`
+```yaml
+# --- Base Model & Tokenizer Configuration ---
+base_model: allenai/Olmo-3-1025-7B
+trust_remote_code: true
+hub_model_id: Auditt/O37BB          # Push the model to the Hugging Face Hub
+chat_template_jinja: /workspace/data/model-output/chat_template.jinja    # Uses the template defined in tokenizer_config.json
+
+# --- Dataset Configuration ---
+# Assuming a standard conversation format (ShareGPT/ChatML style)
+datasets:
+  - path: dataset-tfs-mk-IMP-SOS-processed-olmo3-think.jsonl
+    type: chat_template
+    field_messages: messages    # The top-level key containing the list
+    message_field_role: role            # The key inside the list for 'user'/'assistant'
+    message_field_content: content      # The key inside the list for the actual text
+    
+    # 4. MAP YOUR ROLES
+    # The keys (left) are what Axolotl expects. 
+    # The values (right) are what exist in your raw JSONL file.
+    roles:
+      user: ["user"]
+      assistant: ["assistant"]
+      system: ["system"]
+      
+    # 5. SUPERVISION
+    # This ensures loss is calculated ONLY on the "assistant" turns.
+    roles_to_train: ["assistant"]
+    
+val_set_size: 0.1                   # 10% Validation, 90% Training
+dataset_prepared_path: last_run_prepared
+
+# --- Training Strategy ---
+sequence_len: 60000                 # Max sequence length
+sample_packing: true                # Efficiently packs samples to fill sequence_len
+pad_to_sequence_len: true
+
+# Supervision Settings
+train_on_inputs: false              # False = Mask User prompts (Supervise Assistant only)
+group_by_length: false              # Usually false when sample_packing is true
+
+# --- Hyperparameters & Training Loop ---
+num_epochs: 2
+micro_batch_size: 1                 # Keep small due to 60k context
+gradient_accumulation_steps: 4      # Adjust based on desired global batch size
+learning_rate: 0.00001
+optimizer: adamw_torch
+
+# --- Distributed Training & Memory ---
+context_parallel_size: 2            # Splits the 60k sequence across 2 GPUs
+gradient_checkpointing: true        # Essential for 60k context
+flash_attention: true               # Essential for speed/memory at this length
+
+# --- Logging & Evaluation ---
+logging_steps: 1                    # Log training loss every step
+evals_per_epoch: 1                  # Run eval 1 times per epoch (roughly)
+#eval_strategy: epoch
+#save_strategy: epoch                # Save checkpoint at end of epoch
+#wandb_project: olmo3-finetune       # Optional: Weights & Biases logging
+#wandb_entity: your-entity           # Optional
+output_dir: /workspace/data/model-output-base
+
+# --- Precision ---
+bf16: true                          # Bfloat16 is recommended for OLMo
+fp16: false
+tf32: true
+
+tokens:                             # Add these to the tokenizer
+  - "𐔲"
+  - "𐔾"
+  - "〉"
+  - "𝜎"
+  - "⋁"
+  - "𐕠"
+  - "𐕜"
+  - "𐔸"
+  - "∧"
+  - "≥"
+  - "𐕟"
+  - "𐕖"
+  - "⟂"
+  - "𐕏"
+  - "⋀"
+  - "𐕣"
+  - "𐕃"
+  - "𐕙"
+  - "𐕕"
+  - "χ"
+  - "𐕊"
+  - "〈"
+  - "𐕐"
+  - "𐔻"
+  - "𐕀"
+  - "𐔳"
+  - "≠"
+  - "𐔷"
+  - "≤"
+  - "𐕞"
+  - "𐔱"
+  - "𐕂"
+  - "↦"
+  - "𐕎"
+  - "→"
+  - "𐕛"
+  - "𐔰"
+  - "ε"
+
+```
+
+</details><br>
+
+# O37BB
+
+This model is a fine-tuned version of [allenai/Olmo-3-1025-7B](https://huggingface.co/allenai/Olmo-3-1025-7B) on the dataset-tfs-mk-IMP-SOS-processed-olmo3-think.jsonl dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0019
+- Memory/max Active (gib): 85.95
+- Memory/max Allocated (gib): 82.72
+- Memory/device Reserved (gib): 93.36
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### 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: 2
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 8
+- total_eval_batch_size: 2
+- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_steps: 10
+- training_steps: 348
+
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Active (gib) | Allocated (gib) | Reserved (gib) |
+|:-------------:|:------:|:----:|:---------------:|:------------:|:---------------:|:--------------:|
+| No log        | 0      | 0    | 1.0680          | 58.72        | 55.5            | 65.44          |
+| 0.0647        | 0.9943 | 174  | 0.0021          | 85.95        | 82.72           | 106.04         |
+| 0.0296        | 1.9943 | 348  | 0.0019          | 85.95        | 82.72           | 93.36          |
+
+
+### Framework versions
+
+- Transformers 4.57.0
+- Pytorch 2.7.1+cu126
+- Datasets 4.0.0
+- Tokenizers 0.22.1

model-00001-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:388261a50fbf0eeb4bfe70a569864034136371be5e1eadf296c1c47a754a04ac
+oid sha256:ddfd5c4ee683c28f1c85313fb8c78b300713de4dcbbc0dd015c8e0c63f2841b3
 size 4970296272

model-00002-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4bd68cd0fe72c155079429d0dbebea051320beb1cca26dca7addffb189880c45
+oid sha256:52f8aba6952a7d77fb080415fa01951f7edd404bbf35599e132dbab018acf98f
 size 4981161496

model-00003-of-00003.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4cd8746be0e467fc71540028f00a460140058054a59b338275cc02ce874c724a
+oid sha256:137732105bebe521f2b2ed041974ad2e7097921a82fee9d4417152df112603f5
 size 4645228536