From 0185095402687011db803cb3a5db75a64b9cdd4a Mon Sep 17 00:00:00 2001 From: Alexander Pozharskii Date: Sun, 24 Sep 2023 17:11:19 +0400 Subject: [PATCH] Use `adapter_name` for `get_gptq_peft_model` with `train_mode=True` --- auto_gptq/utils/peft_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_gptq/utils/peft_utils.py b/auto_gptq/utils/peft_utils.py index 9342cbd..46850d0 100644 --- a/auto_gptq/utils/peft_utils.py +++ b/auto_gptq/utils/peft_utils.py @@ -402,7 +402,7 @@ def get_gptq_peft_model( with hijack_peft_mappings(): try: if train_mode: - peft_model = get_peft_model(model.model, peft_config) + peft_model = get_peft_model(model.model, peft_config, adapter_name=adapter_name) else: peft_model = PeftModel.from_pretrained(model.model, model_id, adapter_name) except: