From 6ddb6c2f8ce4154282b3ff18f2861e332a8060a7 Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Sun, 15 Jan 2023 14:50:24 +0100 Subject: [PATCH] Update mapping.py --- src/peft/mapping.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/peft/mapping.py b/src/peft/mapping.py index 46fa815..9ec027f 100644 --- a/src/peft/mapping.py +++ b/src/peft/mapping.py @@ -4,7 +4,7 @@ from .peft_model import ( PeftModelForSequenceClassification, PeftModelForTokenClassification, ) -from .tuners import LoRAConfig, PrefixTuningConfig, PromptEncoderConfig, PromptTuningConfig +from .tuners import LoraConfig, PrefixTuningConfig, PromptEncoderConfig, PromptTuningConfig from .utils import PeftType @@ -19,7 +19,7 @@ PEFT_TYPE_TO_CONFIG_MAPPING = { "PROMPT_TUNING": PromptTuningConfig, "PREFIX_TUNING": PrefixTuningConfig, "P_TUNING": PromptEncoderConfig, - "LORA": LoRAConfig, + "LORA": LoraConfig, } TRANSFORMERS_MODELS_TO_LORA_TARGET_MODULES_MAPPING = {