Run 5. Outer Step 5. Inner Step 2.

config.json

@@ -1,21 +1,20 @@
 {
-  "_name_or_path": "zeras141a/d8",
-  "activation_function": "gelu_new",
+  "_name_or_path": "distributed/llama-1b",
   "all_reduce_scores": {
     "0": "NON_PARTICIPATING",
-    "1": "NON_PARTICIPATING",
+    "1": "SUCCESS",
     "10": "NON_PARTICIPATING",
     "100": "NON_PARTICIPATING",
     "101": "NON_PARTICIPATING",
     "102": "NON_PARTICIPATING",
     "103": "NON_PARTICIPATING",
-    "104": "NON_PARTICIPATING",
+    "104": "SUCCESS",
     "105": "NON_PARTICIPATING",
     "106": "NON_PARTICIPATING",
     "107": "NON_PARTICIPATING",
     "108": "NON_PARTICIPATING",
     "109": "NON_PARTICIPATING",
-    "11": "NON_PARTICIPATING",
+    "11": "SUCCESS",
     "110": "NON_PARTICIPATING",
     "111": "NON_PARTICIPATING",
     "112": "NON_PARTICIPATING",
@@ -24,7 +23,7 @@
     "115": "NON_PARTICIPATING",
     "116": "NON_PARTICIPATING",
     "117": "NON_PARTICIPATING",
-    "118": "NON_PARTICIPATING",
+    "118": "SUCCESS",
     "119": "NON_PARTICIPATING",
     "12": "SUCCESS",
     "120": "NON_PARTICIPATING",
@@ -33,7 +32,7 @@
     "123": "NON_PARTICIPATING",
     "124": "NON_PARTICIPATING",
     "125": "SUCCESS",
-    "126": "NON_PARTICIPATING",
+    "126": "SUCCESS",
     "127": "NON_PARTICIPATING",
     "128": "NON_PARTICIPATING",
     "129": "NON_PARTICIPATING",
@@ -57,7 +56,7 @@
     "145": "NON_PARTICIPATING",
     "146": "NON_PARTICIPATING",
     "147": "NON_PARTICIPATING",
-    "148": "NON_PARTICIPATING",
+    "148": "SUCCESS",
     "149": "NON_PARTICIPATING",
     "15": "SUCCESS",
     "150": "NON_PARTICIPATING",
@@ -70,15 +69,15 @@
     "157": "NON_PARTICIPATING",
     "158": "NON_PARTICIPATING",
     "159": "NON_PARTICIPATING",
-    "16": "NON_PARTICIPATING",
+    "16": "SUCCESS",
     "160": "NON_PARTICIPATING",
-    "161": "SUCCESS",
-    "162": "SUCCESS",
+    "161": "NON_PARTICIPATING",
+    "162": "NON_PARTICIPATING",
     "163": "NON_PARTICIPATING",
     "164": "NON_PARTICIPATING",
     "165": "NON_PARTICIPATING",
     "166": "NON_PARTICIPATING",
-    "167": "NON_PARTICIPATING",
+    "167": "SUCCESS",
     "168": "NON_PARTICIPATING",
     "169": "NON_PARTICIPATING",
     "17": "SUCCESS",
@@ -86,17 +85,17 @@
     "171": "NON_PARTICIPATING",
     "172": "NON_PARTICIPATING",
     "173": "NON_PARTICIPATING",
-    "174": "SUCCESS",
+    "174": "NON_PARTICIPATING",
     "175": "NON_PARTICIPATING",
     "176": "NON_PARTICIPATING",
-    "177": "NON_PARTICIPATING",
+    "177": "SUCCESS",
     "178": "NON_PARTICIPATING",
     "179": "NON_PARTICIPATING",
     "18": "SUCCESS",
     "180": "NON_PARTICIPATING",
     "181": "NON_PARTICIPATING",
     "182": "NON_PARTICIPATING",
-    "183": "SUCCESS",
+    "183": "NON_PARTICIPATING",
     "184": "NON_PARTICIPATING",
     "185": "NON_PARTICIPATING",
     "186": "NON_PARTICIPATING",
@@ -106,7 +105,7 @@
     "19": "SUCCESS",
     "190": "NON_PARTICIPATING",
     "191": "NON_PARTICIPATING",
-    "192": "SUCCESS",
+    "192": "NON_PARTICIPATING",
     "193": "NON_PARTICIPATING",
     "194": "NON_PARTICIPATING",
     "195": "NON_PARTICIPATING",
@@ -121,11 +120,11 @@
     "202": "NON_PARTICIPATING",
     "203": "NON_PARTICIPATING",
     "204": "NON_PARTICIPATING",
-    "205": "SUCCESS",
+    "205": "NON_PARTICIPATING",
     "206": "NON_PARTICIPATING",
     "207": "NON_PARTICIPATING",
     "208": "NON_PARTICIPATING",
-    "209": "NON_PARTICIPATING",
+    "209": "SUCCESS",
     "21": "SUCCESS",
     "210": "NON_PARTICIPATING",
     "211": "NON_PARTICIPATING",
@@ -133,22 +132,22 @@
     "213": "NON_PARTICIPATING",
     "214": "NON_PARTICIPATING",
     "215": "NON_PARTICIPATING",
-    "216": "SUCCESS",
-    "217": "SUCCESS",
+    "216": "NON_PARTICIPATING",
+    "217": "NON_PARTICIPATING",
     "218": "NON_PARTICIPATING",
-    "219": "SUCCESS",
+    "219": "NON_PARTICIPATING",
     "22": "SUCCESS",
     "220": "NON_PARTICIPATING",
     "221": "NON_PARTICIPATING",
     "222": "NON_PARTICIPATING",
-    "223": "NON_PARTICIPATING",
+    "223": "SUCCESS",
     "224": "NON_PARTICIPATING",
     "225": "NON_PARTICIPATING",
     "226": "NON_PARTICIPATING",
     "227": "NON_PARTICIPATING",
     "228": "NON_PARTICIPATING",
     "229": "NON_PARTICIPATING",
-    "23": "NON_PARTICIPATING",
+    "23": "SUCCESS",
     "230": "NON_PARTICIPATING",
     "231": "NON_PARTICIPATING",
     "232": "NON_PARTICIPATING",
@@ -167,54 +166,54 @@
     "244": "NON_PARTICIPATING",
     "245": "NON_PARTICIPATING",
     "246": "NON_PARTICIPATING",
-    "247": "SUCCESS",
+    "247": "NON_PARTICIPATING",
     "248": "NON_PARTICIPATING",
-    "249": "NON_PARTICIPATING",
+    "249": "SUCCESS",
     "25": "SUCCESS",
-    "250": "NON_PARTICIPATING",
+    "250": "SUCCESS",
     "251": "NON_PARTICIPATING",
     "252": "NON_PARTICIPATING",
     "253": "NON_PARTICIPATING",
     "254": "NON_PARTICIPATING",
     "255": "NON_PARTICIPATING",
-    "26": "SUCCESS",
+    "26": "NON_PARTICIPATING",
     "27": "SUCCESS",
-    "28": "NON_PARTICIPATING",
+    "28": "SUCCESS",
     "29": "SUCCESS",
-    "3": "NON_PARTICIPATING",
-    "30": "NON_PARTICIPATING",
+    "3": "SUCCESS",
+    "30": "SUCCESS",
     "31": "SUCCESS",
-    "32": "NON_PARTICIPATING",
+    "32": "SUCCESS",
     "33": "NON_PARTICIPATING",
     "34": "SUCCESS",
     "35": "SUCCESS",
-    "36": "NON_PARTICIPATING",
-    "37": "NON_PARTICIPATING",
+    "36": "SUCCESS",
+    "37": "SUCCESS",
     "38": "SUCCESS",
-    "39": "NON_PARTICIPATING",
+    "39": "SUCCESS",
     "4": "SUCCESS",
     "40": "SUCCESS",
     "41": "SUCCESS",
-    "42": "NON_PARTICIPATING",
-    "43": "NON_PARTICIPATING",
+    "42": "SUCCESS",
+    "43": "SUCCESS",
     "44": "SUCCESS",
     "45": "SUCCESS",
-    "46": "NON_PARTICIPATING",
+    "46": "SUCCESS",
     "47": "SUCCESS",
     "48": "NON_PARTICIPATING",
     "49": "SUCCESS",
-    "5": "NON_PARTICIPATING",
-    "50": "NON_PARTICIPATING",
+    "5": "SUCCESS",
+    "50": "SUCCESS",
     "51": "SUCCESS",
-    "52": "SUCCESS",
-    "53": "NON_PARTICIPATING",
+    "52": "NON_PARTICIPATING",
+    "53": "SUCCESS",
     "54": "SUCCESS",
-    "55": "NON_PARTICIPATING",
+    "55": "SUCCESS",
     "56": "SUCCESS",
-    "57": "NON_PARTICIPATING",
+    "57": "SUCCESS",
     "58": "NON_PARTICIPATING",
     "59": "SUCCESS",
-    "6": "NON_PARTICIPATING",
+    "6": "SUCCESS",
     "60": "SUCCESS",
     "61": "SUCCESS",
     "62": "SUCCESS",
@@ -223,8 +222,8 @@
     "65": "SUCCESS",
     "66": "NON_PARTICIPATING",
     "67": "SUCCESS",
-    "68": "NON_PARTICIPATING",
-    "69": "NON_PARTICIPATING",
+    "68": "SUCCESS",
+    "69": "SUCCESS",
     "7": "SUCCESS",
     "70": "SUCCESS",
     "71": "SUCCESS",
@@ -232,71 +231,63 @@
     "73": "SUCCESS",
     "74": "SUCCESS",
     "75": "SUCCESS",
-    "76": "NON_PARTICIPATING",
+    "76": "SUCCESS",
     "77": "SUCCESS",
-    "78": "NON_PARTICIPATING",
-    "79": "NON_PARTICIPATING",
+    "78": "SUCCESS",
+    "79": "SUCCESS",
     "8": "SUCCESS",
     "80": "SUCCESS",
-    "81": "NON_PARTICIPATING",
+    "81": "SUCCESS",
     "82": "SUCCESS",
-    "83": "NON_PARTICIPATING",
+    "83": "SUCCESS",
     "84": "SUCCESS",
     "85": "SUCCESS",
     "86": "NON_PARTICIPATING",
     "87": "SUCCESS",
-    "88": "NON_PARTICIPATING",
+    "88": "SUCCESS",
     "89": "SUCCESS",
-    "9": "SUCCESS",
+    "9": "NON_PARTICIPATING",
     "90": "SUCCESS",
     "91": "SUCCESS",
     "92": "SUCCESS",
-    "93": "NON_PARTICIPATING",
-    "94": "NON_PARTICIPATING",
+    "93": "SUCCESS",
+    "94": "SUCCESS",
     "95": "SUCCESS",
-    "96": "NON_PARTICIPATING",
+    "96": "SUCCESS",
     "97": "SUCCESS",
     "98": "SUCCESS",
     "99": "SUCCESS"
   },
   "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": [
-    5691782,
-    5691802
+    5706753,
+    5706783
   ],
-  "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": 3,
-  "inner_steps": 0,
-  "last_allreduce_block": 5686307,
-  "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": 2,
+  "intermediate_size": 5632,
+  "last_allreduce_block": 5706512,
+  "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
+}

inner_optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0030fe7d6fcb420eb1f087734bcdb207ddce6227b502f24bacf3431ba2996d87
-size 8081782026
+oid sha256:9b2f716c04c0cba3f9e685730d4c55dcff56c8e5f46c8654043587768087b8f4
+size 8800564012

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fee9bf97993d35d4946a69727e970171f3e841d888f56adbcc35bd583d0b206e
-size 4040701744
+oid sha256:0a8b6a63448eac9dab06bbe49e61ab0c65c0840cb52eaac91743fe2eb7edd164
+size 4400216536