From 56c19e623c17f292ec59eccda412ab998ad423b1 Mon Sep 17 00:00:00 2001 From: FartyPants Date: Wed, 14 Jun 2023 17:29:42 -0400 Subject: [PATCH] Add LORA name instead of "default" in PeftModel (#2689) --- modules/LoRA.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/LoRA.py b/modules/LoRA.py index e74bfff0..403a89ad 100644 --- a/modules/LoRA.py +++ b/modules/LoRA.py @@ -77,7 +77,7 @@ def add_lora_to_model(lora_names): elif shared.args.load_in_8bit: params['device_map'] = {'': 0} - shared.model = PeftModel.from_pretrained(shared.model, Path(f"{shared.args.lora_dir}/{lora_names[0]}"), **params) + shared.model = PeftModel.from_pretrained(shared.model, Path(f"{shared.args.lora_dir}/{lora_names[0]}"),adapter_name=lora_names[0], **params) for lora in lora_names[1:]: shared.model.load_adapter(Path(f"{shared.args.lora_dir}/{lora}"), lora)