ybelkada
/

tiny-random-LlamaForCausalLM-GQA

Text Generation

text-generation-inference

Model card Files Files and versions

Update config.json

#2

by ybelkada - opened Nov 17, 2023

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -8,10 +8,10 @@
   "bos_token_id": 0,
   "eos_token_id": 1,
   "hidden_act": "silu",
-  "hidden_size": 128,
   "initializer_range": 0.02,
-  "intermediate_size": 64,
-  "max_position_embeddings": 2048,
   "model_type": "llama",
   "num_attention_heads": 64,
   "num_hidden_layers": 2,

   "bos_token_id": 0,
   "eos_token_id": 1,
   "hidden_act": "silu",
+  "hidden_size": 8192,
   "initializer_range": 0.02,
+  "intermediate_size": 28672,
+  "max_position_embeddings": 4096,
   "model_type": "llama",
   "num_attention_heads": 64,
   "num_hidden_layers": 2,