TaylorAI
/

Flash-Llama-30M-4001

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

andersonbcdefg commited on Sep 4, 2023

Commit

1bb4900

·

1 Parent(s): 09f89b0

Upload LlamaForCausalLM

Files changed (2) hide show

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

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "TaylorAI/Flash-Llama-220M",
   "architectures": [
     "LlamaForCausalLM"
   ],
@@ -9,13 +9,13 @@
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",
-  "hidden_size": 768,
   "initializer_range": 0.02,
-  "intermediate_size": 2048,
   "max_position_embeddings": 32768,
   "model_type": "llama",
   "num_attention_heads": 12,
-  "num_hidden_layers": 24,
   "num_key_value_heads": 12,
   "pad_token_id": 0,
   "pretraining_tp": 1,

 {
+  "_name_or_path": "TaylorAI/Flash-Llama-30M",
   "architectures": [
     "LlamaForCausalLM"
   ],
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",
+  "hidden_size": 384,
   "initializer_range": 0.02,
+  "intermediate_size": 1024,
   "max_position_embeddings": 32768,
   "model_type": "llama",
   "num_attention_heads": 12,
+  "num_hidden_layers": 4,
   "num_key_value_heads": 12,
   "pad_token_id": 0,
   "pretraining_tp": 1,

pytorch_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:42f849718d2dbcbf442e57d0b15dd27068653ae7f224b5cf5769049f7b3bb68b
+size 126642150