config = LoraConfig( | |
task_type=TaskType.CAUSAL_LM, | |
target_modules=["q_proj", "k_proj", "v_proj", "o_proj", "gate_proj", "up_proj", "down_proj"], | |
inference_mode=False, # 训练模式 | |
r=64, # Lora 秩 | |
lora_alpha=128, # Lora alaph,具体作用参见 Lora 原理 | |
lora_dropout=0.1# Dropout 比例 | |
) |