unionai
/

pythia-1B-deduped-wikipedia

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nielsbantilan commited on Jun 12, 2023

Commit

073797c

•

1 Parent(s): 0e0b629

Upload GPTNeoXForCausalLM

Files changed (2) hide show

config.json +2 -2
pytorch_model.bin +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "EleutherAI/pythia-1B-deduped",
   "architectures": [
     "GPTNeoXForCausalLM"
   ],
@@ -18,7 +18,7 @@
   "rotary_emb_base": 10000,
   "rotary_pct": 0.25,
   "tie_word_embeddings": false,
-  "torch_dtype": "float16",
   "transformers_version": "4.29.2",
   "use_cache": true,
   "use_parallel_residual": true,

 {
+  "_name_or_path": "/tmp/flyteav9zvdg1/local_flytekit/3dc3fdcbe70c98829ce1a05b3cd622c0",
   "architectures": [
     "GPTNeoXForCausalLM"
   ],
   "rotary_emb_base": 10000,
   "rotary_pct": 0.25,
   "tie_word_embeddings": false,
+  "torch_dtype": "float32",
   "transformers_version": "4.29.2",
   "use_cache": true,
   "use_parallel_residual": true,

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:3ff095431efbef5d7a3b7e02f6a0888af48d5e1c2fe8de04c43c1c810c7e1b49
+size 4114316253