Mufcruz commited on
Commit
0ce81bd
1 Parent(s): ba3d102

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "model_type": "llama",
3
- "architectures": ["LlamaForSequenceClassification"],
4
- "hidden_size": 768,
5
- "num_attention_heads": 12,
6
- "num_hidden_layers": 12,
7
- "vocab_size": 50257
8
  }
 
1
  {
2
  "model_type": "llama",
3
+ "pipeline_tag": "feature-extraction",
4
+ "hidden_size": 4096,
5
+ "num_attention_heads": 32,
6
+ "num_hidden_layers": 32,
7
+ "vocab_size": 32000
8
  }