Upload folder using huggingface_hub

README.md

@@ -1,3 +1,60 @@
 ---
 license: apache-2.0
+base_model: state-spaces/mamba-2.8b-slimpj
+datasets:
+- HuggingFaceH4/ultrachat_200k
+model-index:
+- name: mamba-2.8b-ultrachat
+  results: []
 ---
+
+# mamba-2.8b-ultrachat
+
+This model is a fine-tuned version of [state-spaces/mamba-2.8b-slimpj](https://huggingface.co/state-spaces/mamba-2.8b-slimpj) on the [HuggingFaceH4/ultrachat_200k](https://huggingface.co/datasets/HuggingFaceH4/ultrachat_200k) dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.1858
+
+## 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: 2e-05
+- train_batch_size: 4
+- eval_batch_size: 4
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 8
+- gradient_accumulation_steps: 16
+- total_train_batch_size: 512
+- total_eval_batch_size: 32
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- num_epochs: 1
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 2.0106        | 0.0   | 1    | 1.9092          |
+| 1.1783        | 0.62  | 250  | 1.1858          |
+
+
+### Framework versions
+
+- Transformers 4.35.0
+- Pytorch 2.1.1+cu121
+- Datasets 2.14.6
+- Tokenizers 0.14.1

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 0.62,
+    "eval_loss": 1.185780644416809,
+    "eval_runtime": 399.119,
+    "eval_samples": 23110,
+    "eval_samples_per_second": 57.903,
+    "eval_steps_per_second": 1.811,
+    "train_loss": 1.2281424560546874,
+    "train_runtime": 11653.8648,
+    "train_samples": 207865,
+    "train_samples_per_second": 17.837,
+    "train_steps_per_second": 0.035
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 0.62,
+    "eval_loss": 1.185780644416809,
+    "eval_runtime": 399.119,
+    "eval_samples": 23110,
+    "eval_samples_per_second": 57.903,
+    "eval_steps_per_second": 1.811
+}

pytorch_model.bin

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

special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<|endoftext|>",
+  "eos_token": "<|endoftext|>",
+  "pad_token": "<|endoftext|>",
+  "unk_token": "<|endoftext|>"
+}

tokenizer_config.json

@@ -0,0 +1,213 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<|padding|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50254": {
+      "content": "                        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50255": {
+      "content": "                       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50256": {
+      "content": "                      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50257": {
+      "content": "                     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50258": {
+      "content": "                    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50259": {
+      "content": "                   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50260": {
+      "content": "                  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50261": {
+      "content": "                 ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50262": {
+      "content": "                ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50263": {
+      "content": "               ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50264": {
+      "content": "              ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50265": {
+      "content": "             ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50266": {
+      "content": "            ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50267": {
+      "content": "           ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50268": {
+      "content": "          ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50269": {
+      "content": "         ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50270": {
+      "content": "        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50271": {
+      "content": "       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50272": {
+      "content": "      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50273": {
+      "content": "     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50274": {
+      "content": "    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50275": {
+      "content": "   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50276": {
+      "content": "  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "bos_token": "<|endoftext|>",
+  "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n'  + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|endoftext|>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<|endoftext|>",
+  "tokenizer_class": "GPTNeoXTokenizer",
+  "unk_token": "<|endoftext|>"
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 0.62,
+    "train_loss": 1.2281424560546874,
+    "train_runtime": 11653.8648,
+    "train_samples": 207865,
+    "train_samples_per_second": 17.837,
+    "train_steps_per_second": 0.035
+}

trainer_state.json

@@ -0,0 +1,350 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.6157635467980296,
+  "eval_steps": 500,
+  "global_step": 250,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.9999700625010444e-05,
+      "loss": 2.0106,
+      "step": 1
+    },
+    {
+      "epoch": 0.0,
+      "eval_loss": 1.9091932773590088,
+      "eval_runtime": 396.1093,
+      "eval_samples_per_second": 58.342,
+      "eval_steps_per_second": 1.825,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.999251652147735e-05,
+      "loss": 1.7041,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.997007728639956e-05,
+      "loss": 1.5057,
+      "step": 10
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9932715879473385e-05,
+      "loss": 1.3651,
+      "step": 15
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9880488219356086e-05,
+      "loss": 1.3209,
+      "step": 20
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.981347247496222e-05,
+      "loss": 1.2837,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.973176894846855e-05,
+      "loss": 1.2683,
+      "step": 30
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.963549992519223e-05,
+      "loss": 1.2609,
+      "step": 35
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9524809490566878e-05,
+      "loss": 1.2493,
+      "step": 40
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.939986331449053e-05,
+      "loss": 1.2467,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.926084840336821e-05,
+      "loss": 1.238,
+      "step": 50
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.910797282022027e-05,
+      "loss": 1.2322,
+      "step": 55
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.894146537327533e-05,
+      "loss": 1.2183,
+      "step": 60
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.8761575273514005e-05,
+      "loss": 1.2114,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8568571761675893e-05,
+      "loss": 1.2217,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8362743705288127e-05,
+      "loss": 1.216,
+      "step": 75
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.814439916631857e-05,
+      "loss": 1.2176,
+      "step": 80
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.791386494010081e-05,
+      "loss": 1.2022,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.7671486066220965e-05,
+      "loss": 1.2047,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7417625312098453e-05,
+      "loss": 1.219,
+      "step": 95
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7152662630033506e-05,
+      "loss": 1.2135,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.6876994588534234e-05,
+      "loss": 1.2006,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.659103377877423e-05,
+      "loss": 1.1977,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.629520819706912e-05,
+      "loss": 1.2003,
+      "step": 115
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.598996060429634e-05,
+      "loss": 1.2023,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.56757478632168e-05,
+      "loss": 1.2081,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5353040254690396e-05,
+      "loss": 1.1959,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 1.5022320773808612e-05,
+      "loss": 1.1897,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 1.4684084406997903e-05,
+      "loss": 1.1958,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 1.4338837391175582e-05,
+      "loss": 1.1818,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 1.3987096456067236e-05,
+      "loss": 1.1937,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 1.3629388050819547e-05,
+      "loss": 1.186,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 1.3266247556066122e-05,
+      "loss": 1.1851,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 1.2898218482625606e-05,
+      "loss": 1.185,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 1.252585165803135e-05,
+      "loss": 1.1964,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 1.2149704402110243e-05,
+      "loss": 1.1899,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 1.1770339692844484e-05,
+      "loss": 1.1901,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 1.1388325323764889e-05,
+      "loss": 1.179,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 1.1004233054136726e-05,
+      "loss": 1.1816,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 1.0618637753210086e-05,
+      "loss": 1.1929,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 1.0232116539815558e-05,
+      "loss": 1.1995,
+      "step": 200
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 9.845247918592937e-06,
+      "loss": 1.1818,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 9.458610914145826e-06,
+      "loss": 1.1844,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 9.072784204417995e-06,
+      "loss": 1.1863,
+      "step": 215
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 8.688345254588579e-06,
+      "loss": 1.1884,
+      "step": 220
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 8.305869452782446e-06,
+      "loss": 1.1918,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 7.92592924888925e-06,
+      "loss": 1.1751,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 7.549093297780133e-06,
+      "loss": 1.1807,
+      "step": 235
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 7.175925608204428e-06,
+      "loss": 1.1807,
+      "step": 240
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 6.806984698640202e-06,
+      "loss": 1.1865,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 6.442822761362015e-06,
+      "loss": 1.1783,
+      "step": 250
+    },
+    {
+      "epoch": 0.62,
+      "eval_loss": 1.1858137845993042,
+      "eval_runtime": 399.1577,
+      "eval_samples_per_second": 57.897,
+      "eval_steps_per_second": 1.811,
+      "step": 250
+    },
+    {
+      "epoch": 0.62,
+      "step": 250,
+      "total_flos": 0.0,
+      "train_loss": 1.2281424560546874,
+      "train_runtime": 11653.8648,
+      "train_samples_per_second": 17.837,
+      "train_steps_per_second": 0.035
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 406,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "total_flos": 0.0,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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