trl-internal-testing
/

tiny-BloomForCausalLM

Text Generation

text-generation-inference

Model card Files Files and versions

qgallouedec HF Staff commited on Sep 23

Commit

1ee1bae

·

verified ·

1 Parent(s): 2c3c38f

Upload BloomForCausalLM

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -7,7 +7,7 @@
   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,
   "bos_token_id": 1,
-  "dtype": "float32",
   "eos_token_id": 2,
   "hidden_dropout": 0.0,
   "hidden_size": 8,

   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,
   "bos_token_id": 1,
+  "dtype": "bfloat16",
   "eos_token_id": 2,
   "hidden_dropout": 0.0,
   "hidden_size": 8,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:bbc5925eb2a3b3f12d145b11219c296b4441d548e2aaf3aadb5eee54535b5248
+size 4020600