trl-internal-testing
/

tiny-Gemma2ForCausalLM

Text Generation

text-generation-inference

Model card Files Files and versions

qgallouedec HF Staff commited on Sep 23

Commit

95dcc47

·

verified ·

1 Parent(s): 47430c5

Upload Gemma2ForCausalLM

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -7,7 +7,7 @@
   "attn_logit_softcapping": 50.0,
   "bos_token_id": 2,
   "cache_implementation": "hybrid",
-  "dtype": "float32",
   "eos_token_id": [
     1,
     107

   "attn_logit_softcapping": 50.0,
   "bos_token_id": 2,
   "cache_implementation": "hybrid",
+  "dtype": "bfloat16",
   "eos_token_id": [
     1,
     107

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:31c16135c340347c6e76077a3655ccf85a95b286cfb37a396b9ce0c2723c8c9f
-size 8397584

 version https://git-lfs.github.com/spec/v1
+oid sha256:20ab3be5e38d62b2df985362578f6d02b17eb217c1d2466fc7b54f2c8178e42f
+size 4200088