trl-internal-testing
/

tiny-GemmaForCausalLM

Text Generation

text-generation-inference

Model card Files Files and versions

qgallouedec HF Staff commited on Sep 24, 2025

Commit

1c491b0

·

verified ·

1 Parent(s): f06ea78

Upload GemmaForCausalLM

Files changed (2) hide show

config.json +2 -3
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -5,10 +5,10 @@
   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 2,
-  "dtype": "bfloat16",
   "eos_token_id": 1,
   "head_dim": 256,
-  "hidden_act": "gelu",
   "hidden_size": 8,
   "initializer_range": 0.02,
   "intermediate_size": 32,
@@ -19,7 +19,6 @@
   "num_key_value_heads": 2,
   "pad_token_id": 0,
   "rms_norm_eps": 1e-06,
-  "rope_scaling": null,
   "rope_theta": 10000.0,
   "transformers_version": "4.57.0.dev0",
   "use_cache": true,

   "attention_bias": false,
   "attention_dropout": 0.0,
   "bos_token_id": 2,
+  "dtype": "float32",
   "eos_token_id": 1,
   "head_dim": 256,
+  "hidden_act": "gelu_pytorch_tanh",
   "hidden_size": 8,
   "initializer_range": 0.02,
   "intermediate_size": 32,
   "num_key_value_heads": 2,
   "pad_token_id": 0,
   "rms_norm_eps": 1e-06,
   "rope_theta": 10000.0,
   "transformers_version": "4.57.0.dev0",
   "use_cache": true,

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:54e4940594272d092f2de6601e4c7aa68be6bcce8c5e4e3b9486c576bc5b7a36
-size 4199576

 version https://git-lfs.github.com/spec/v1
+oid sha256:92c1fd0f66cac6eac1e5a9bbd2ebe05744abee8f4003437fd56cc95db67a0b82
+size 8397008