loulou2
/

tiny_llama

Text Generation

text-generation-inference

Model card Files Files and versions

loulou2 commited on Aug 22

Commit

da3c230

·

verified ·

1 Parent(s): fe1db9a

feat: change nb layers

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -19,7 +19,7 @@
   "mlp_bias": false,
   "model_type": "llama",
   "num_attention_heads": 32,
-  "num_hidden_layers": 16,
   "num_key_value_heads": 8,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,

   "mlp_bias": false,
   "model_type": "llama",
   "num_attention_heads": 32,
+  "num_hidden_layers": 1,
   "num_key_value_heads": 8,
   "pretraining_tp": 1,
   "rms_norm_eps": 1e-05,