TaylorAI
/

Flash-Llama-30M-6001

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

andersonbcdefg commited on Sep 4, 2023

Commit

18b1d96

·

1 Parent(s): c8ea81f

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:63c79465711a37f9c8bd490f6ca6f54f7c873ae06086d8e669a2b2c0301179df
-size 876309369

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