Update config.json
Browse files- config.json +2 -2
config.json
CHANGED
@@ -3,9 +3,9 @@
|
|
3 |
"peft_model_id": "mhhmm/codegen-6B-LoRA",
|
4 |
"_name_or_path": "codegen-6B-mono",
|
5 |
"architectures": [
|
6 |
-
"
|
7 |
],
|
8 |
-
"model_type": "
|
9 |
"pad_token_id": 1023,
|
10 |
"fan_in_fan_out": false,
|
11 |
"inference_mode": true,
|
|
|
3 |
"peft_model_id": "mhhmm/codegen-6B-LoRA",
|
4 |
"_name_or_path": "codegen-6B-mono",
|
5 |
"architectures": [
|
6 |
+
"AutoModelForCausalLM"
|
7 |
],
|
8 |
+
"model_type": "peft",
|
9 |
"pad_token_id": 1023,
|
10 |
"fan_in_fan_out": false,
|
11 |
"inference_mode": true,
|