trl-internal-testing
/

tiny-GPTNeoXForCausalLM

Text Generation

text-generation-inference

Model card Files Files and versions

qgallouedec HF Staff commited on Sep 23, 2025

Commit

dd97e84

·

verified ·

1 Parent(s): 7ea7aa3

Upload GPTNeoXForCausalLM

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "classifier_dropout": 0.1,
-  "dtype": "float32",
   "eos_token_id": 0,
   "hidden_act": "gelu",
   "hidden_dropout": 0.0,

   "attention_dropout": 0.0,
   "bos_token_id": 0,
   "classifier_dropout": 0.1,
+  "dtype": "bfloat16",
   "eos_token_id": 0,
   "hidden_act": "gelu",
   "hidden_dropout": 0.0,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:771d85716b39e56a7902501550ff035433bb540fdfcc373c4fad813c76283fb7
+size 1616272