support for DoRA w/ PEFT (#1363)

This commit is contained in:
Wing Lian
2024-03-05 21:20:15 -05:00
committed by GitHub
parent 37657473c8
commit 0cfdb2c90c
3 changed files with 16 additions and 2 deletions

View File

@@ -830,6 +830,8 @@ def load_lora(model, cfg, inference=False, config_only=False):
if loftq_bits:
lora_config_kwargs["loftq_config"] = LoftQConfig(loftq_bits=loftq_bits)
lora_config_kwargs["init_lora_weights"] = "loftq"
if cfg.peft_use_dora:
lora_config_kwargs["use_dora"] = cfg.peft_use_dora
lora_config = LoraConfig(
r=cfg.lora_r,