Upload GemmaForCausalLM

config.json

@@ -9,7 +9,6 @@
   "eos_token_id": 213,
   "head_dim": 256,
   "hidden_act": "gelu",
-  "hidden_activation": null,
   "hidden_size": 2048,
   "initializer_range": 0.02,
   "intermediate_size": 16384,
@@ -19,11 +18,9 @@
   "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.2",
   "use_cache": false,

generation_config.json

@@ -1,7 +1,6 @@
 {
   "_from_model_config": true,
   "bos_token_id": 204,
-  "do_sample": true,
   "eos_token_id": 213,
   "pad_token_id": 0,
   "transformers_version": "4.38.2",

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:193170c2a9e3fab8aaf8dd1ae5679132f49d3bd37da80fa2bf942c5f956b9740
+oid sha256:fc1f86703050ded5b192be9e16d6761e81b1c1af3f18a50abfab801eec3edba9
 size 4945242264

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4a264f04546f91efc6a6c01f6de51bc3c86b32d6597bad5e2a2412877629a71c
-size 1115697720
+oid sha256:c3f0c8a3c7e91201e5150a38c693534bbd1ac64f5362affbb238db7971f542b2
+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",