Update config.json
Browse files- config.json +7 -1
config.json
CHANGED
@@ -3,8 +3,14 @@
|
|
3 |
"peft_model_id": "mhhmm/codegen-6B-LoRA",
|
4 |
"_name_or_path": "codegen-6B-mono",
|
5 |
"architectures": [
|
6 |
-
"
|
7 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
"model_type": "peft",
|
9 |
"pad_token_id": 1023,
|
10 |
"fan_in_fan_out": false,
|
|
|
3 |
"peft_model_id": "mhhmm/codegen-6B-LoRA",
|
4 |
"_name_or_path": "codegen-6B-mono",
|
5 |
"architectures": [
|
6 |
+
"AutoModelForCausalLM"
|
7 |
],
|
8 |
+
"tokenizer_class": "GPT2Tokenizer",
|
9 |
+
"attention_probs_dropout_prob": 0.0,
|
10 |
+
"attn_pdrop": 0.0,
|
11 |
+
"bos_token_id": 0,
|
12 |
+
"embd_pdrop": 0.0,
|
13 |
+
"eos_token_id": 0,
|
14 |
"model_type": "peft",
|
15 |
"pad_token_id": 1023,
|
16 |
"fan_in_fan_out": false,
|