TaylorAI
/

Flash-Llama-30M-14001

Text Generation

text-generation-inference

Model card Files Files and versions

andersonbcdefg commited on Sep 4, 2023

Commit

0b59fe0

·

1 Parent(s): b2e018e

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:d2a9d6145319d84f089cbb82ac89c9f9a1c1286b26027f86828b3b1fbc37f4f7
-size 876309369

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