simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Use adapter_name for get_gptq_peft_model with train_mode=True

Browse source
This commit is contained in:
Alexander Pozharskii 2023-09-24 17:11:19 +04:00
parent 06e071e68e
commit 0185095402
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
auto_gptq/utils/peft_utils.py
View file

@ -402,7 +402,7 @@ def get_gptq_peft_model(
with hijack_peft_mappings(): with hijack_peft_mappings():
try: try:
if train_mode: 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: else:
peft_model = PeftModel.from_pretrained(model.model, model_id, adapter_name) peft_model = PeftModel.from_pretrained(model.model, model_id, adapter_name)
except: except: