frank6626 commited on
Commit
cce07d3
·
verified ·
1 Parent(s): e2047ad

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -3,7 +3,10 @@
3
  "ArcticLSTMSpeculatorPreTrainedModel"
4
  ],
5
  "base_model_name_or_path": "meta-llama/Llama-3.1-70B-Instruct",
6
- "base_model_arch": "LlamaForCausalLM",
 
 
 
7
  "input_hidden_dim": 8192,
8
  "inner_dim": "4096",
9
  "proj_dim": "4096",
 
3
  "ArcticLSTMSpeculatorPreTrainedModel"
4
  ],
5
  "base_model_name_or_path": "meta-llama/Llama-3.1-70B-Instruct",
6
+ "base_model_archs": [
7
+ "LlamaForCausalLM",
8
+ "LlamaSwiftKVForCausalLM"
9
+ ],
10
  "input_hidden_dim": 8192,
11
  "inner_dim": "4096",
12
  "proj_dim": "4096",