support loftq

This commit is contained in:
hiyouga
2023-12-12 22:47:06 +08:00
parent ada0e536c9
commit 6219dfbd93
5 changed files with 42 additions and 19 deletions

View File

@@ -55,6 +55,10 @@ class LoraArguments:
Phi-1.5 choices: [\"Wqkv\", \"out_proj\", \"fc1\", \"fc2\"], \
Others choices: the same as LLaMA."}
)
loftq_init: Optional[bool] = field(
default=False,
metadata={"help": "Use LoftQ initialization for quantized LoRA fine-tuning."}
)
resume_lora_training: Optional[bool] = field(
default=True,
metadata={"help": "Whether to resume training from the last LoRA weights or create new weights after merging them."}