| { | |
| "base_model_name_or_path": "/workspace/WizardLM-7b", | |
| "bias": "all", | |
| "fan_in_fan_out": false, | |
| "inference_mode": true, | |
| "init_lora_weights": true, | |
| "lora_alpha": 64, | |
| "lora_dropout": 0.1, | |
| "modules_to_save": null, | |
| "peft_type": "LORA", | |
| "r": 8, | |
| "target_modules": [ | |
| "q_proj", | |
| "k_proj", | |
| "v_proj", | |
| "out_proj", | |
| "fc_in", | |
| "fc_out", | |
| "wte" | |
| ], | |
| "task_type": "CAUSAL_LM" | |
| } |