{ "_name_or_path": "/tmp/models/tmpovpu48eq/tmpklks3m1s", "activation_function": "gelu_new", "all_reduce_scores": { "0": "NON_PARTICIPATING", "1": "NON_PARTICIPATING", "10": "SUCCESS", "100": "NON_PARTICIPATING", "101": "SUCCESS", "102": "NON_PARTICIPATING", "103": "NON_PARTICIPATING", "104": "NON_PARTICIPATING", "105": "SUCCESS", "106": "SUCCESS", "107": "SUCCESS", "108": "SUCCESS", "109": "NON_PARTICIPATING", "11": "NON_PARTICIPATING", "110": "NON_PARTICIPATING", "111": "NON_PARTICIPATING", "112": "SUCCESS", "113": "SUCCESS", "114": "SUCCESS", "115": "NON_PARTICIPATING", "116": "SUCCESS", "117": "NON_PARTICIPATING", "118": "NON_PARTICIPATING", "119": "SUCCESS", "12": "NON_PARTICIPATING", "120": "SUCCESS", "121": "SUCCESS", "122": "NON_PARTICIPATING", "123": "NON_PARTICIPATING", "124": "NON_PARTICIPATING", "125": "NON_PARTICIPATING", "126": "SUCCESS", "127": "NON_PARTICIPATING", "128": "NON_PARTICIPATING", "129": "SUCCESS", "13": "NON_PARTICIPATING", "130": "SUCCESS", "131": "NON_PARTICIPATING", "132": "SUCCESS", "133": "SUCCESS", "134": "NON_PARTICIPATING", "135": "SUCCESS", "136": "NON_PARTICIPATING", "137": "NON_PARTICIPATING", "138": "NON_PARTICIPATING", "139": "NON_PARTICIPATING", "14": "NON_PARTICIPATING", "140": "SUCCESS", "141": "SUCCESS", "142": "NON_PARTICIPATING", "143": "NON_PARTICIPATING", "144": "NON_PARTICIPATING", "145": "NON_PARTICIPATING", "146": "NON_PARTICIPATING", "147": "NON_PARTICIPATING", "148": "SUCCESS", "149": "NON_PARTICIPATING", "15": "NON_PARTICIPATING", "150": "NON_PARTICIPATING", "151": "SUCCESS", "152": "NON_PARTICIPATING", "153": "NON_PARTICIPATING", "154": "SUCCESS", "155": "SUCCESS", "156": "NON_PARTICIPATING", "157": "NON_PARTICIPATING", "158": "NON_PARTICIPATING", "159": "SUCCESS", "16": "NON_PARTICIPATING", "160": "SUCCESS", "161": "SUCCESS", "162": "SUCCESS", "163": "SUCCESS", "164": "NON_PARTICIPATING", "165": "NON_PARTICIPATING", "166": "SUCCESS", "167": "NON_PARTICIPATING", "168": "SUCCESS", "169": "NON_PARTICIPATING", "17": "SUCCESS", "170": "NON_PARTICIPATING", "171": "SUCCESS", "172": "NON_PARTICIPATING", "173": "NON_PARTICIPATING", "174": "SUCCESS", "175": "SUCCESS", "176": "SUCCESS", "177": "SUCCESS", "178": "NON_PARTICIPATING", "179": "NON_PARTICIPATING", "18": "NON_PARTICIPATING", "180": "NON_PARTICIPATING", "181": "NON_PARTICIPATING", "182": "NON_PARTICIPATING", "183": "SUCCESS", "184": "NON_PARTICIPATING", "185": "NON_PARTICIPATING", "186": "SUCCESS", "187": "SUCCESS", "188": "NON_PARTICIPATING", "189": "SUCCESS", "19": "SUCCESS", "190": "SUCCESS", "191": "NON_PARTICIPATING", "192": "SUCCESS", "193": "NON_PARTICIPATING", "194": "NON_PARTICIPATING", "195": "SUCCESS", "196": "NON_PARTICIPATING", "197": "SUCCESS", "198": "NON_PARTICIPATING", "199": "NON_PARTICIPATING", "2": "SUCCESS", "20": "NON_PARTICIPATING", "200": "SUCCESS", "201": "NON_PARTICIPATING", "202": "NON_PARTICIPATING", "203": "NON_PARTICIPATING", "204": "NON_PARTICIPATING", "205": "SUCCESS", "206": "NON_PARTICIPATING", "207": "NON_PARTICIPATING", "208": "NON_PARTICIPATING", "209": "NON_PARTICIPATING", "21": "NON_PARTICIPATING", "210": "NON_PARTICIPATING", "211": "SUCCESS", "212": "NON_PARTICIPATING", "213": "NON_PARTICIPATING", "214": "NON_PARTICIPATING", "215": "NON_PARTICIPATING", "216": "NON_PARTICIPATING", "217": "SUCCESS", "218": "SUCCESS", "219": "SUCCESS", "22": "SUCCESS", "220": "SUCCESS", "221": "NON_PARTICIPATING", "222": "NON_PARTICIPATING", "223": "NON_PARTICIPATING", "224": "NON_PARTICIPATING", "225": "NON_PARTICIPATING", "226": "SUCCESS", "227": "NON_PARTICIPATING", "228": "NON_PARTICIPATING", "229": "NON_PARTICIPATING", "23": "SUCCESS", "230": "SUCCESS", "231": "NON_PARTICIPATING", "232": "NON_PARTICIPATING", "233": "SUCCESS", "234": "SUCCESS", "235": "NON_PARTICIPATING", "236": "NON_PARTICIPATING", "237": "SUCCESS", "238": "NON_PARTICIPATING", "239": "NON_PARTICIPATING", "24": "SUCCESS", "240": "NON_PARTICIPATING", "241": "NON_PARTICIPATING", "242": "NON_PARTICIPATING", "243": "NON_PARTICIPATING", "244": "NON_PARTICIPATING", "245": "NON_PARTICIPATING", "246": "SUCCESS", "247": "SUCCESS", "248": "NON_PARTICIPATING", "249": "NON_PARTICIPATING", "25": "NON_PARTICIPATING", "250": "SUCCESS", "251": "SUCCESS", "252": "SUCCESS", "253": "NON_PARTICIPATING", "254": "SUCCESS", "255": "NON_PARTICIPATING", "26": "NON_PARTICIPATING", "27": "SUCCESS", "28": "NON_PARTICIPATING", "29": "NON_PARTICIPATING", "3": "SUCCESS", "30": "SUCCESS", "31": "NON_PARTICIPATING", "32": "SUCCESS", "33": "NON_PARTICIPATING", "34": "NON_PARTICIPATING", "35": "SUCCESS", "36": "NON_PARTICIPATING", "37": "NON_PARTICIPATING", "38": "NON_PARTICIPATING", "39": "SUCCESS", "4": "NON_PARTICIPATING", "40": "NON_PARTICIPATING", "41": "NON_PARTICIPATING", "42": "NON_PARTICIPATING", "43": "SUCCESS", "44": "NON_PARTICIPATING", "45": "SUCCESS", "46": "NON_PARTICIPATING", "47": "SUCCESS", "48": "NON_PARTICIPATING", "49": "NON_PARTICIPATING", "5": "NON_PARTICIPATING", "50": "SUCCESS", "51": "NON_PARTICIPATING", "52": "NON_PARTICIPATING", "53": "SUCCESS", "54": "SUCCESS", "55": "SUCCESS", "56": "NON_PARTICIPATING", "57": "NON_PARTICIPATING", "58": "NON_PARTICIPATING", "59": "NON_PARTICIPATING", "6": "SUCCESS", "60": "NON_PARTICIPATING", "61": "NON_PARTICIPATING", "62": "NON_PARTICIPATING", "63": "SUCCESS", "64": "NON_PARTICIPATING", "65": "NON_PARTICIPATING", "66": "SUCCESS", "67": "SUCCESS", "68": "NON_PARTICIPATING", "69": "NON_PARTICIPATING", "7": "SUCCESS", "70": "NON_PARTICIPATING", "71": "SUCCESS", "72": "NON_PARTICIPATING", "73": "SUCCESS", "74": "NON_PARTICIPATING", "75": "SUCCESS", "76": "SUCCESS", "77": "NON_PARTICIPATING", "78": "SUCCESS", "79": "NON_PARTICIPATING", "8": "SUCCESS", "80": "NON_PARTICIPATING", "81": "NON_PARTICIPATING", "82": "SUCCESS", "83": "NON_PARTICIPATING", "84": "SUCCESS", "85": "NON_PARTICIPATING", "86": "SUCCESS", "87": "SUCCESS", "88": "NON_PARTICIPATING", "89": "NON_PARTICIPATING", "9": "NON_PARTICIPATING", "90": "SUCCESS", "91": "NON_PARTICIPATING", "92": "NON_PARTICIPATING", "93": "NON_PARTICIPATING", "94": "NON_PARTICIPATING", "95": "NON_PARTICIPATING", "96": "NON_PARTICIPATING", "97": "NON_PARTICIPATING", "98": "NON_PARTICIPATING", "99": "SUCCESS" }, "architectures": [ "GPTOptim" ], "attn_pdrop": 0.1, "auto_map": { "AutoConfig": "configuration_gpt_optimized.GPTOptimConfig", "AutoModelForCausalLM": "distributed/optimized-gpt2-2b--modeling_gpt_optimized.GPTOptim" }, "block_size": 1024, "bos_token_id": 50256, "embd_pdrop": 0.1, "eos_token_id": 50256, "initializer_range": 0.02, "layer_norm_epsilon": 1e-05, "model_type": "gpt_optimized", "n_embd": 1904, "n_head": 34, "n_inner": null, "n_layer": 44, "n_positions": 1024, "quantization_config": { "transformer.h.attn.c_attn": { "offload_meta": false, "scale_quant_params": null, "weight_quant_params": { "axis": 0, "channel_wise": true, "group_size": 64, "nbits": 4, "optimize": true, "round_zero": true, "view_as_float": false }, "zero_quant_params": { "channel_wise": false, "group_size": null, "nbits": 8, "optimize": false } }, "transformer.h.attn.c_proj": { "offload_meta": false, "scale_quant_params": null, "weight_quant_params": { "axis": 0, "channel_wise": true, "group_size": 64, "nbits": 4, "optimize": true, "round_zero": true, "view_as_float": false }, "zero_quant_params": { "channel_wise": false, "group_size": null, "nbits": 8, "optimize": false } }, "transformer.h.mlp.c_fc": { "offload_meta": false, "scale_quant_params": null, "weight_quant_params": { "axis": 0, "channel_wise": true, "group_size": 64, "nbits": 4, "optimize": true, "round_zero": true, "view_as_float": false }, "zero_quant_params": { "channel_wise": false, "group_size": null, "nbits": 8, "optimize": false } }, "transformer.h.mlp.c_proj": { "offload_meta": false, "scale_quant_params": null, "weight_quant_params": { "axis": 0, "channel_wise": true, "group_size": 64, "nbits": 4, "optimize": true, "round_zero": true, "view_as_float": false }, "zero_quant_params": { "channel_wise": false, "group_size": null, "nbits": 8, "optimize": false } } }, "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, "torch_dtype": "float32", "transformers_version": "4.44.0", "use_cache": true, "vocab_size": 50257 }