Upload GemmaForCausalLM

config.json

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "/temp/auto_model",
   "architectures": [
-    "LlamaForCausalLM"
+    "GemmaForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
@@ -13,16 +13,14 @@
   "initializer_range": 0.02,
   "intermediate_size": 16384,
   "max_position_embeddings": 8192,
-  "model_type": "llama",
+  "model_type": "gemma",
   "num_attention_heads": 8,
   "num_hidden_layers": 18,
   "num_key_value_heads": 1,
   "pad_token_id": 0,
-  "pretraining_tp": 1,
   "rms_norm_eps": 1e-06,
   "rope_scaling": null,
   "rope_theta": 10000.0,
-  "tie_word_embeddings": false,
   "torch_dtype": "bfloat16",
   "transformers_version": "4.38.1",
   "use_cache": false,

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd9f8e2790094a12f4767d4da5dcfb2ec8102eeb35ae92235c397f3517051b73
+oid sha256:7bd337f8a79f92599ee546f383305ed02c147b970b5363504554c675511371e7
 size 4945242264

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b46c468e0ec495474000c690e4a99f100c4da860646f8af1526acc95f6878a9
-size 1115697720
+oid sha256:a8c0f9b83ed793650a8a1ada3afda6fc8c9735f9d9bf8e421f98c2e3490d4343
+size 67121608

model.safetensors.index.json

@@ -1,9 +1,8 @@
 {
   "metadata": {
-    "total_size": 6060920832
+    "total_size": 5012344832
   },
   "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",