from peft import LoraConfig

import config

def load_adapter(target_modules):
    peft_config = LoraConfig(
        lora_alpha=config.LORA_ALPHA,
        lora_dropout=config.LORA_DROPOUT,
        r = config.LORA_RANK,
        bias="none",
        task_type=config.TASK_TYPE,
        target_modules=target_modules
    )
    return peft_config