Upload MPTForCausalLM

config.json

@@ -0,0 +1,54 @@
+{
+  "_name_or_path": "replit/replit-code-v1-3b",
+  "architectures": [
+    "MPTForCausalLM"
+  ],
+  "attn_config": {
+    "alibi": true,
+    "alibi_bias_max": 8,
+    "attn_impl": "torch",
+    "attn_pdrop": 0,
+    "attn_type": "multihead_attention",
+    "attn_uses_sequence_id": false,
+    "clip_qkv": null,
+    "prefix_lm": false,
+    "qk_ln": false,
+    "softmax_scale": null
+  },
+  "auto_map": {
+    "AutoConfig": "replit/replit-code-v1-3b--configuration_mpt.MPTConfig",
+    "AutoModelForCausalLM": "replit/replit-code-v1-3b--modeling_mpt.MPTForCausalLM"
+  },
+  "d_model": 2560,
+  "emb_pdrop": 0,
+  "embedding_fraction": 1.0,
+  "eos_token_id": 1,
+  "expansion_ratio": 4,
+  "init_config": {
+    "emb_init_std": null,
+    "emb_init_uniform_lim": null,
+    "fan_mode": "fan_in",
+    "init_div_is_residual": true,
+    "init_gain": 0,
+    "init_nonlinearity": "relu",
+    "init_std": 0.02,
+    "name": "kaiming_normal_",
+    "verbose": 0
+  },
+  "init_device": "cpu",
+  "learned_pos_emb": true,
+  "logit_scale": null,
+  "max_seq_len": 2048,
+  "model_type": "mpt",
+  "n_heads": 32,
+  "n_layers": 32,
+  "no_bias": true,
+  "norm_type": "low_precision_layernorm",
+  "resid_pdrop": 0,
+  "tokenizer_name": "replit/replit-code-v1-3b",
+  "torch_dtype": "float32",
+  "transformers_version": "4.31.0",
+  "use_cache": true,
+  "verbose": 0,
+  "vocab_size": 32768
+}

generation_config.json

@@ -0,0 +1,5 @@
+{
+  "_from_model_config": true,
+  "eos_token_id": 1,
+  "transformers_version": "4.31.0"
+}

pytorch_model-00001-of-00002.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48a1c104c1d733cd0f8a3dc2d4d17267d8589796c627124f37e5e9900e4a8a18
+size 9983144925

pytorch_model-00002-of-00002.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5807999c95d363b472268a90e95c7a98a7d575f2a9c43c707cdcd581d3d1e146
+size 419464289

pytorch_model.bin.index.json

@@ -0,0 +1,201 @@
+{
+  "metadata": {
+    "total_size": 10402539520
+  },
+  "weight_map": {
+    "transformer.blocks.0.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.0.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.0.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.0.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.0.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.0.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.1.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.10.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.11.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.12.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.13.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.14.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.15.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.16.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.17.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.18.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.19.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.2.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.20.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.21.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.22.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.23.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.24.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.25.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.26.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.27.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.28.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.29.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.3.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.30.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.30.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.30.ffn.down_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.30.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.30.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.30.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.31.attn.Wqkv.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.31.attn.out_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.31.ffn.down_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.31.ffn.up_proj.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.31.norm_1.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.31.norm_2.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.blocks.4.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.4.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.4.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.4.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.4.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.4.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.5.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.6.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.7.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.8.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.attn.Wqkv.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.attn.out_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.ffn.down_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.ffn.up_proj.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.norm_1.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.blocks.9.norm_2.weight": "pytorch_model-00001-of-00002.bin",
+    "transformer.norm_f.weight": "pytorch_model-00002-of-00002.bin",
+    "transformer.wte.weight": "pytorch_model-00001-of-00002.bin"
+  }
+}