Upload GPTNeoXForCausalLM

README.md

@@ -1,15 +1,15 @@
 ---
+base_model: EleutherAI/pythia-410m
+library_name: transformers
+license: other
 tags:
 - autotrain
 - text-generation-inference
 - text-generation
-library_name: transformers
-base_model: EleutherAI/pythia-410m
 widget:
-  - messages:
-      - role: user
-        content: What is your favorite condiment?
-license: other
+- messages:
+  - role: user
+    content: What is your favorite condiment?
 ---
 
 # Model Trained Using AutoTrain

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "EleutherAI/pythia-410m",
+  "_name_or_path": "/home/nicknames/SN80/ai-factory-arxiv-model-improved",
   "architectures": [
     "GPTNeoXForCausalLM"
   ],
@@ -25,7 +25,7 @@
   "rotary_pct": 0.25,
   "tie_word_embeddings": false,
   "torch_dtype": "bfloat16",
-  "transformers_version": "4.48.0",
+  "transformers_version": "4.44.1",
   "use_cache": true,
   "use_parallel_residual": true,
   "vocab_size": 50277

generation_config.json

@@ -2,6 +2,6 @@
   "_from_model_config": true,
   "bos_token_id": 0,
   "eos_token_id": 0,
-  "transformers_version": "4.48.0",
+  "transformers_version": "4.44.1",
   "use_cache": false
 }