Upload LlamaForCausalLM

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "meta-llama/Llama-3.2-3B",
+  "_name_or_path": "output_3b/stage_finetuning",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -28,8 +28,7 @@
       "registry_requires_subclass": false,
       "sparsity_structure": "0:0",
       "targets": []
-    },
-    "version": "0.8.1"
+    }
   },
   "rms_norm_eps": 1e-05,
   "rope_scaling": {
@@ -41,8 +40,8 @@
   },
   "rope_theta": 500000.0,
   "tie_word_embeddings": true,
-  "torch_dtype": "bfloat16",
+  "torch_dtype": "float16",
   "transformers_version": "4.47.1",
   "use_cache": true,
   "vocab_size": 128256
-}
+}

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b594353ab63527420015e922d5b183eedca772fd537493519c0f9670f681b10
-size 4965799096
+oid sha256:f43ccecb9aaf86b39956a79df25d6e3ea4231ab76edf7a9dc7cdcfaec3367ecc
+size 4965798912

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f1584bdafefe77d92afd776bc37ad189100c155571de707ebaa495c811d6f707
-size 2247734992
+oid sha256:ae07f73d2eb03b1f6b204294fe125d5c663be80e285a22246f1f4859cdce720d
+size 1459729880

model.safetensors.index.json

@@ -1,9 +1,8 @@
 {
   "metadata": {
-    "total_size": 7213504512
+    "total_size": 6425499648
   },
   "weight_map": {
-    "lm_head.weight": "model-00002-of-00002.safetensors",
     "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
     "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
     "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",