JaphetHernandez commited on
Commit
9212192
1 Parent(s): e2acd7a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -6
config.json CHANGED
@@ -1,8 +1,11 @@
1
  {
2
- "task": "text-generation",
3
- "model_type": "llama",
4
- "vocab_size": 50257,
5
- "hidden_size": 2048,
6
- "num_attention_heads": 16,
7
- "num_hidden_layers": 12
 
 
 
8
  }
 
1
  {
2
+ "model_type": "llama",
3
+ "task": "text-generation",
4
+ "vocab_size": 50257,
5
+ "hidden_size": 2048,
6
+ "auto_map": {
7
+ "AutoConfig": "JaphetHernandez/llama_retrained--config",
8
+ "AutoModel": "JaphetHernandez/llama_retrained--model",
9
+ "AutoModelForCausalLM": "JaphetHernandez/llama_retrained--model"
10
+ }
11
  }