armaniii
/

llama-argument-classification

Text Classification

text-generation

Model card Files Files and versions

armaniii commited on Apr 13, 2024

Commit

4fa504f

·

verified ·

1 Parent(s): ad8ca23

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
-{
   "_name_or_path": "meta-llama/Llama-2-7b-hf",
   "architectures": [
-    "LlamaForSequenceClassification"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
@@ -11,7 +11,7 @@
   "hidden_size": 4096,
   "initializer_range": 0.02,
   "intermediate_size": 11008,
-  "max_position_embeddings": 2048,
   "model_type": "llama",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,

+LlamaConfig {
   "_name_or_path": "meta-llama/Llama-2-7b-hf",
   "architectures": [
+    "LlamaForCausalLM"
   ],
   "attention_bias": false,
   "attention_dropout": 0.0,
   "hidden_size": 4096,
   "initializer_range": 0.02,
   "intermediate_size": 11008,
+  "max_position_embeddings": 4096,
   "model_type": "llama",
   "num_attention_heads": 32,
   "num_hidden_layers": 32,