kcduece commited on
Commit
a5af0f0
·
1 Parent(s): cce58ea

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +16 -6
config.json CHANGED
@@ -1,8 +1,18 @@
1
  {
2
- "architectures": [
3
- "GPT2LMHeadModel"
4
- ],
5
- "model_type": "gpt2",
6
- "tokenizer_class": "GPT2Tokenizer",
7
- "vocab_size": 50257
 
 
 
 
 
 
 
 
 
 
8
  }
 
1
  {
2
+ "base_model_name_or_path": "decapoda-research/llama-7b-hf",
3
+ "bias": "none",
4
+ "enable_lora": null,
5
+ "fan_in_fan_out": false,
6
+ "inference_mode": true,
7
+ "lora_alpha": 16,
8
+ "lora_dropout": 0.05,
9
+ "merge_weights": false,
10
+ "modules_to_save": null,
11
+ "peft_type": "LORA",
12
+ "r": 4,
13
+ "target_modules": [
14
+ "q_proj",
15
+ "v_proj"
16
+ ],
17
+ "task_type": "CAUSAL_LM"
18
  }