Upload HF_GPTForCausalLM

config.json

@@ -0,0 +1,29 @@
+{
+  "architectures": [
+    "HF_GPTForCausalLM"
+  ],
+  "bos_token_id": 1,
+  "conv_size": 4,
+  "eos_token_id": 2,
+  "expand_k": 1.0,
+  "expand_v": 1.0,
+  "hidden_size": null,
+  "mode": "fused_chunk",
+  "model_type": "hf_gpt",
+  "name": "LLaMA_1.3B_SWA",
+  "norm_eps": 1e-05,
+  "norm_first": false,
+  "num_heads": 4,
+  "pad_token_id": 0,
+  "qk_activation": "silu",
+  "qk_norm": "l2",
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.43.2",
+  "use_beta": true,
+  "use_cache": true,
+  "use_gate": false,
+  "use_output_norm": true,
+  "use_short_conv": true,
+  "vocab_size": 32000
+}

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "pad_token_id": 0,
+  "transformers_version": "4.43.2"
+}

model-00001-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38d81e1df2808640f04f55968eefd7e347f766eaacda961a4e70d2d6b2daa513
+size 4966682664

model-00002-of-00002.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:878183e7c9da719f8a5204e9761b996ee29ffadb834f964672fdf7242f69059a
+size 719614248

model.safetensors.index.json

@@ -0,0 +1,150 @@
+{
+  "metadata": {
+    "total_size": 5686281216
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00002-of-00002.safetensors",
+    "transformer.embeddings.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.0.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.1.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.10.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.11.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.12.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.13.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.14.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.15.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.16.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.17.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.18.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.18.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.18.mlp.swiglu.w1.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.18.mlp.swiglu.w2.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.18.mlp.swiglu.w3.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.18.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.18.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.19.attn.attn.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.attn.proj.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.mlp.swiglu.w1.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.mlp.swiglu.w2.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.mlp.swiglu.w3.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.norm_1.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.19.norm_2.weight": "model-00002-of-00002.safetensors",
+    "transformer.h.2.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.2.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.3.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.4.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.5.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.6.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.7.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.8.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.attn.attn.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.attn.proj.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.mlp.swiglu.w1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.mlp.swiglu.w2.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.mlp.swiglu.w3.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.norm_1.weight": "model-00001-of-00002.safetensors",
+    "transformer.h.9.norm_2.weight": "model-00001-of-00002.safetensors",
+    "transformer.ln_f.weight": "model-00002-of-00002.safetensors"
+  }
+}