support for rslora (#1387) [skip ci]

This commit is contained in:
Wing Lian
2024-03-10 20:49:45 -04:00
committed by GitHub
parent 3fd8093717
commit 7659c001aa
2 changed files with 3 additions and 0 deletions

View File

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