Joetib
/

pythia-finetuned-5-steps

Text Generation

Generated from Trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Joetib commited on Nov 3, 2023

Commit

20d811b

•

1 Parent(s): 1c8ec68

Upload GPTNeoXForCausalLM

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "EleutherAI/pythia-410M",
   "architectures": [
     "GPTNeoXForCausalLM"
   ],
@@ -21,7 +21,7 @@
   "rotary_emb_base": 10000,
   "rotary_pct": 0.25,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": "4.35.0",
   "use_cache": true,
   "use_parallel_residual": true,

 {
+  "_name_or_path": "pythia-finetuned-5-steps",
   "architectures": [
     "GPTNeoXForCausalLM"
   ],
   "rotary_emb_base": 10000,
   "rotary_pct": 0.25,
   "tie_word_embeddings": false,
+  "torch_dtype": "float16",
   "transformers_version": "4.35.0",
   "use_cache": true,
   "use_parallel_residual": true,

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:297f1e680e970efd0a78c049ccc3256f43607db6b23a48389f8df3886094ec85
+size 810701896