trl-internal-testing
/

tiny-GPT2LMHeadModel

Text Generation

text-generation-inference

Model card Files Files and versions

qgallouedec HF Staff commited on Sep 23

Commit

2525b3c

·

verified ·

1 Parent(s): 96bba53

Upload model

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -5,7 +5,7 @@
   ],
   "attn_pdrop": 0.1,
   "bos_token_id": 50256,
-  "dtype": "float32",
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,

   ],
   "attn_pdrop": 0.1,
   "bos_token_id": 50256,
+  "dtype": "bfloat16",
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
   "initializer_range": 0.02,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:72a86752357d5c0f3994ba5098cd26e69c3bcd3bff589c92c67cb780b41abbdf
+size 826536