Run 5. Outer Step 150. Inner Step 1.

config.json

@@ -1,32 +1,31 @@
 {
-  "_name_or_path": "distributed/optimized-gpt2-1b",
-  "activation_function": "gelu_new",
+  "_name_or_path": "distributed/llama-1b",
   "all_reduce_scores": {
-    "0": "NON_PARTICIPATING",
+    "0": "NOT_ALIVE",
     "1": "NON_PARTICIPATING",
     "10": "NON_PARTICIPATING",
-    "100": "NON_PARTICIPATING",
-    "101": "NON_PARTICIPATING",
+    "100": "NOT_ALIVE",
+    "101": "NOT_ALIVE",
     "102": "NON_PARTICIPATING",
     "103": "NON_PARTICIPATING",
-    "104": "NON_PARTICIPATING",
+    "104": "NOT_ALIVE",
     "105": "NON_PARTICIPATING",
     "106": "NON_PARTICIPATING",
     "107": "NON_PARTICIPATING",
-    "108": "NON_PARTICIPATING",
+    "108": "NOT_ALIVE",
     "109": "NON_PARTICIPATING",
     "11": "NON_PARTICIPATING",
     "110": "NON_PARTICIPATING",
-    "111": "NON_PARTICIPATING",
+    "111": "NOT_ALIVE",
     "112": "NON_PARTICIPATING",
     "113": "NON_PARTICIPATING",
     "114": "NON_PARTICIPATING",
     "115": "NON_PARTICIPATING",
     "116": "NON_PARTICIPATING",
     "117": "NON_PARTICIPATING",
-    "118": "NON_PARTICIPATING",
-    "119": "NON_PARTICIPATING",
-    "12": "NON_PARTICIPATING",
+    "118": "NOT_ALIVE",
+    "119": "NOT_ALIVE",
+    "12": "NOT_ALIVE",
     "120": "NON_PARTICIPATING",
     "121": "NON_PARTICIPATING",
     "122": "NON_PARTICIPATING",
@@ -34,12 +33,12 @@
     "124": "NON_PARTICIPATING",
     "125": "NON_PARTICIPATING",
     "126": "NON_PARTICIPATING",
-    "127": "NON_PARTICIPATING",
+    "127": "NOT_ALIVE",
     "128": "NON_PARTICIPATING",
     "129": "NON_PARTICIPATING",
-    "13": "NON_PARTICIPATING",
+    "13": "NOT_ALIVE",
     "130": "NON_PARTICIPATING",
-    "131": "NON_PARTICIPATING",
+    "131": "NOT_ALIVE",
     "132": "NON_PARTICIPATING",
     "133": "NON_PARTICIPATING",
     "134": "NON_PARTICIPATING",
@@ -48,54 +47,54 @@
     "137": "NON_PARTICIPATING",
     "138": "NON_PARTICIPATING",
     "139": "NON_PARTICIPATING",
-    "14": "NON_PARTICIPATING",
-    "140": "NON_PARTICIPATING",
+    "14": "NOT_ALIVE",
+    "140": "NOT_ALIVE",
     "141": "NON_PARTICIPATING",
-    "142": "NON_PARTICIPATING",
+    "142": "NOT_ALIVE",
     "143": "NON_PARTICIPATING",
     "144": "NON_PARTICIPATING",
-    "145": "NON_PARTICIPATING",
-    "146": "NON_PARTICIPATING",
+    "145": "NOT_ALIVE",
+    "146": "NOT_ALIVE",
     "147": "NON_PARTICIPATING",
     "148": "NON_PARTICIPATING",
-    "149": "NON_PARTICIPATING",
-    "15": "NON_PARTICIPATING",
+    "149": "NOT_ALIVE",
+    "15": "NOT_ALIVE",
     "150": "NON_PARTICIPATING",
-    "151": "NON_PARTICIPATING",
+    "151": "NOT_ALIVE",
     "152": "NON_PARTICIPATING",
-    "153": "NON_PARTICIPATING",
+    "153": "NOT_ALIVE",
     "154": "NON_PARTICIPATING",
     "155": "NON_PARTICIPATING",
-    "156": "NON_PARTICIPATING",
-    "157": "NON_PARTICIPATING",
+    "156": "NOT_ALIVE",
+    "157": "NOT_ALIVE",
     "158": "NON_PARTICIPATING",
-    "159": "NON_PARTICIPATING",
-    "16": "NON_PARTICIPATING",
-    "160": "NON_PARTICIPATING",
-    "161": "NON_PARTICIPATING",
+    "159": "NOT_ALIVE",
+    "16": "NOT_ALIVE",
+    "160": "NOT_ALIVE",
+    "161": "NOT_ALIVE",
     "162": "NON_PARTICIPATING",
-    "163": "NON_PARTICIPATING",
+    "163": "NOT_ALIVE",
     "164": "NON_PARTICIPATING",
     "165": "NON_PARTICIPATING",
     "166": "NON_PARTICIPATING",
-    "167": "NON_PARTICIPATING",
-    "168": "NON_PARTICIPATING",
-    "169": "NON_PARTICIPATING",
-    "17": "NON_PARTICIPATING",
-    "170": "NON_PARTICIPATING",
-    "171": "NON_PARTICIPATING",
+    "167": "NOT_ALIVE",
+    "168": "NOT_ALIVE",
+    "169": "NOT_ALIVE",
+    "17": "NOT_ALIVE",
+    "170": "NOT_ALIVE",
+    "171": "NOT_ALIVE",
     "172": "NON_PARTICIPATING",
-    "173": "NON_PARTICIPATING",
+    "173": "NOT_ALIVE",
     "174": "NON_PARTICIPATING",
     "175": "NON_PARTICIPATING",
-    "176": "NON_PARTICIPATING",
+    "176": "NOT_ALIVE",
     "177": "NON_PARTICIPATING",
     "178": "NON_PARTICIPATING",
     "179": "NON_PARTICIPATING",
     "18": "NON_PARTICIPATING",
     "180": "NON_PARTICIPATING",
-    "181": "NON_PARTICIPATING",
-    "182": "NON_PARTICIPATING",
+    "181": "NOT_ALIVE",
+    "182": "NOT_ALIVE",
     "183": "NON_PARTICIPATING",
     "184": "NON_PARTICIPATING",
     "185": "NON_PARTICIPATING",
@@ -107,199 +106,188 @@
     "190": "NON_PARTICIPATING",
     "191": "NON_PARTICIPATING",
     "192": "NON_PARTICIPATING",
-    "193": "NON_PARTICIPATING",
+    "193": "NOT_ALIVE",
     "194": "NON_PARTICIPATING",
-    "195": "NON_PARTICIPATING",
+    "195": "NOT_ALIVE",
     "196": "NON_PARTICIPATING",
-    "197": "NON_PARTICIPATING",
+    "197": "NOT_ALIVE",
     "198": "NON_PARTICIPATING",
     "199": "NON_PARTICIPATING",
     "2": "NON_PARTICIPATING",
-    "20": "NON_PARTICIPATING",
+    "20": "NOT_ALIVE",
     "200": "NON_PARTICIPATING",
-    "201": "NON_PARTICIPATING",
+    "201": "NOT_ALIVE",
     "202": "NON_PARTICIPATING",
-    "203": "NON_PARTICIPATING",
+    "203": "NOT_ALIVE",
     "204": "NON_PARTICIPATING",
-    "205": "NON_PARTICIPATING",
-    "206": "NON_PARTICIPATING",
-    "207": "NON_PARTICIPATING",
+    "205": "NOT_ALIVE",
+    "206": "NOT_ALIVE",
+    "207": "NOT_ALIVE",
     "208": "NON_PARTICIPATING",
-    "209": "NON_PARTICIPATING",
-    "21": "NON_PARTICIPATING",
-    "210": "NON_PARTICIPATING",
+    "209": "NOT_ALIVE",
+    "21": "NOT_ALIVE",
+    "210": "NOT_ALIVE",
     "211": "NON_PARTICIPATING",
     "212": "NON_PARTICIPATING",
     "213": "NON_PARTICIPATING",
     "214": "NON_PARTICIPATING",
     "215": "NON_PARTICIPATING",
     "216": "NON_PARTICIPATING",
-    "217": "NON_PARTICIPATING",
+    "217": "NOT_ALIVE",
     "218": "NON_PARTICIPATING",
-    "219": "NON_PARTICIPATING",
+    "219": "NOT_ALIVE",
     "22": "NON_PARTICIPATING",
     "220": "NON_PARTICIPATING",
     "221": "NON_PARTICIPATING",
     "222": "NON_PARTICIPATING",
     "223": "NON_PARTICIPATING",
-    "224": "NON_PARTICIPATING",
+    "224": "NOT_ALIVE",
     "225": "NON_PARTICIPATING",
     "226": "NON_PARTICIPATING",
     "227": "NON_PARTICIPATING",
     "228": "NON_PARTICIPATING",
     "229": "NON_PARTICIPATING",
     "23": "NON_PARTICIPATING",
-    "230": "NON_PARTICIPATING",
-    "231": "NON_PARTICIPATING",
-    "232": "NON_PARTICIPATING",
+    "230": "NOT_ALIVE",
+    "231": "NOT_ALIVE",
+    "232": "NOT_ALIVE",
     "233": "NON_PARTICIPATING",
-    "234": "NON_PARTICIPATING",
+    "234": "NOT_ALIVE",
     "235": "NON_PARTICIPATING",
     "236": "NON_PARTICIPATING",
     "237": "NON_PARTICIPATING",
     "238": "NON_PARTICIPATING",
-    "239": "NON_PARTICIPATING",
-    "24": "NON_PARTICIPATING",
+    "239": "NOT_ALIVE",
+    "24": "NOT_ALIVE",
     "240": "NON_PARTICIPATING",
-    "241": "NON_PARTICIPATING",
+    "241": "NOT_ALIVE",
     "242": "NON_PARTICIPATING",
     "243": "NON_PARTICIPATING",
     "244": "NON_PARTICIPATING",
     "245": "NON_PARTICIPATING",
-    "246": "NON_PARTICIPATING",
+    "246": "NOT_ALIVE",
     "247": "NON_PARTICIPATING",
-    "248": "NON_PARTICIPATING",
+    "248": "NOT_ALIVE",
     "249": "NON_PARTICIPATING",
-    "25": "NON_PARTICIPATING",
-    "250": "NON_PARTICIPATING",
-    "251": "NON_PARTICIPATING",
-    "252": "NON_PARTICIPATING",
+    "25": "SUCCESS",
+    "250": "NOT_ALIVE",
+    "251": "NOT_ALIVE",
+    "252": "NOT_ALIVE",
     "253": "NON_PARTICIPATING",
-    "254": "NON_PARTICIPATING",
-    "255": "NON_PARTICIPATING",
+    "254": "NOT_ALIVE",
+    "255": "NOT_ALIVE",
     "26": "NON_PARTICIPATING",
-    "27": "NON_PARTICIPATING",
+    "27": "NOT_ALIVE",
     "28": "NON_PARTICIPATING",
-    "29": "NON_PARTICIPATING",
+    "29": "NOT_ALIVE",
     "3": "NON_PARTICIPATING",
-    "30": "NON_PARTICIPATING",
+    "30": "NOT_ALIVE",
     "31": "NON_PARTICIPATING",
     "32": "NON_PARTICIPATING",
-    "33": "NON_PARTICIPATING",
-    "34": "NON_PARTICIPATING",
+    "33": "NOT_ALIVE",
+    "34": "NOT_ALIVE",
     "35": "NON_PARTICIPATING",
-    "36": "NON_PARTICIPATING",
+    "36": "NOT_ALIVE",
     "37": "NON_PARTICIPATING",
-    "38": "NON_PARTICIPATING",
+    "38": "NOT_ALIVE",
     "39": "NON_PARTICIPATING",
     "4": "NON_PARTICIPATING",
     "40": "NON_PARTICIPATING",
     "41": "NON_PARTICIPATING",
-    "42": "NON_PARTICIPATING",
-    "43": "NON_PARTICIPATING",
+    "42": "NOT_ALIVE",
+    "43": "NOT_ALIVE",
     "44": "NON_PARTICIPATING",
     "45": "NON_PARTICIPATING",
     "46": "NON_PARTICIPATING",
     "47": "NON_PARTICIPATING",
-    "48": "NON_PARTICIPATING",
-    "49": "NON_PARTICIPATING",
+    "48": "NOT_ALIVE",
+    "49": "NOT_ALIVE",
     "5": "NON_PARTICIPATING",
     "50": "NON_PARTICIPATING",
     "51": "NON_PARTICIPATING",
     "52": "NON_PARTICIPATING",
     "53": "NON_PARTICIPATING",
-    "54": "NON_PARTICIPATING",
-    "55": "NON_PARTICIPATING",
+    "54": "NOT_ALIVE",
+    "55": "NOT_ALIVE",
     "56": "NON_PARTICIPATING",
-    "57": "NON_PARTICIPATING",
+    "57": "NOT_ALIVE",
     "58": "NON_PARTICIPATING",
-    "59": "NON_PARTICIPATING",
+    "59": "NOT_ALIVE",
     "6": "NON_PARTICIPATING",
-    "60": "NON_PARTICIPATING",
+    "60": "NOT_ALIVE",
     "61": "NON_PARTICIPATING",
     "62": "NON_PARTICIPATING",
     "63": "NON_PARTICIPATING",
-    "64": "NON_PARTICIPATING",
+    "64": "NOT_ALIVE",
     "65": "NON_PARTICIPATING",
     "66": "NON_PARTICIPATING",
-    "67": "NON_PARTICIPATING",
+    "67": "NOT_ALIVE",
     "68": "NON_PARTICIPATING",
-    "69": "NON_PARTICIPATING",
-    "7": "NON_PARTICIPATING",
+    "69": "NOT_ALIVE",
+    "7": "NOT_ALIVE",
     "70": "NON_PARTICIPATING",
     "71": "NON_PARTICIPATING",
     "72": "NON_PARTICIPATING",
     "73": "NON_PARTICIPATING",
-    "74": "NON_PARTICIPATING",
-    "75": "NON_PARTICIPATING",
-    "76": "NON_PARTICIPATING",
+    "74": "NOT_ALIVE",
+    "75": "NOT_ALIVE",
+    "76": "NOT_ALIVE",
     "77": "NON_PARTICIPATING",
     "78": "NON_PARTICIPATING",
     "79": "NON_PARTICIPATING",
-    "8": "NON_PARTICIPATING",
+    "8": "NOT_ALIVE",
     "80": "NON_PARTICIPATING",
-    "81": "NON_PARTICIPATING",
+    "81": "NOT_ALIVE",
     "82": "NON_PARTICIPATING",
-    "83": "NON_PARTICIPATING",
+    "83": "NOT_ALIVE",
     "84": "NON_PARTICIPATING",
-    "85": "NON_PARTICIPATING",
+    "85": "NOT_ALIVE",
     "86": "NON_PARTICIPATING",
-    "87": "NON_PARTICIPATING",
-    "88": "NON_PARTICIPATING",
-    "89": "NON_PARTICIPATING",
+    "87": "NOT_ALIVE",
+    "88": "NOT_ALIVE",
+    "89": "NOT_ALIVE",
     "9": "NON_PARTICIPATING",
-    "90": "NON_PARTICIPATING",
+    "90": "NOT_ALIVE",
     "91": "NON_PARTICIPATING",
-    "92": "NON_PARTICIPATING",
-    "93": "NON_PARTICIPATING",
-    "94": "NON_PARTICIPATING",
-    "95": "NON_PARTICIPATING",
-    "96": "NON_PARTICIPATING",
+    "92": "NOT_ALIVE",
+    "93": "NOT_ALIVE",
+    "94": "NOT_ALIVE",
+    "95": "NOT_ALIVE",
+    "96": "NOT_ALIVE",
     "97": "NON_PARTICIPATING",
-    "98": "NON_PARTICIPATING",
+    "98": "NOT_ALIVE",
     "99": "NON_PARTICIPATING"
   },
   "architectures": [
-    "GPTOptim"
+    "LlamaForCausalLM"
   ],
-  "attn_pdrop": 0.1,
-  "auto_map": {
-    "AutoConfig": "distributed/optimized-gpt2-500m--configuration_gpt_optimized.GPTOptimConfig",
-    "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
-  },
+  "attention_bias": false,
+  "attention_dropout": 0.0,
   "block_list": [
-    5470149,
-    5470153,
-    5470157,
-    5470162,
-    5470166
+    6069837,
+    6069849
   ],
-  "block_size": 1024,
-  "bos_token_id": 50256,
-  "embd_pdrop": 0.1,
-  "eos_token_id": 50256,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
   "initializer_range": 0.02,
-  "inner_step": 129,
-  "inner_steps": 0,
-  "last_allreduce_block": 5450451,
-  "layer_norm_epsilon": 1e-05,
-  "model_type": "gpt_optimized",
-  "n_embd": 1280,
-  "n_head": 32,
-  "n_inner": null,
-  "n_layer": 48,
-  "n_positions": 1024,
-  "reorder_and_upcast_attn": false,
-  "resid_pdrop": 0.1,
-  "scale_attn_by_inverse_layer_idx": false,
-  "scale_attn_weights": true,
-  "summary_activation": null,
-  "summary_first_dropout": 0.1,
-  "summary_proj_to_labels": true,
-  "summary_type": "cls_index",
-  "summary_use_proj": true,
+  "inner_step": 1,
+  "intermediate_size": 5632,
+  "last_allreduce_block": 6062756,
+  "max_position_embeddings": 2048,
+  "mlp_bias": false,
+  "model_type": "llama",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 22,
+  "num_key_value_heads": 4,
+  "pretraining_tp": 1,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 10000.0,
+  "tie_word_embeddings": false,
   "torch_dtype": "float32",
   "transformers_version": "4.39.3",
-  "use_cache": true,
-  "vocab_size": 50257
+  "use_cache": false,
+  "vocab_size": 32000
 }

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "transformers_version": "4.39.3",
+  "use_cache": false
+}

gradients.pt

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

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9c76567b4f8117f5aaff62609232dad2e7e87408fb20658dcfd0089e119dfb29
-size 8081782026
+oid sha256:6bb5ceddb77ebe382ada0cfe2fdb46780197f027a33b6585b1fd7b3108e6aaa4
+size 8800564076

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9cc123cdf6f4d87e5d735134bce93f5b030196fb27b312fbeefb28dfd1336d33
-size 4040701744
+oid sha256:ef6da77306659acb1efd185596abfda76c3ed075239bb542a0dfc38b126c2cb5
+size 4400216536