o Ni&@sUddlmZddlmZmZmZddlZddlmZer)ddl m Z ddl m Z iZ ded <iZd ed <iZd ed <iZd ed<d"ddZ   d#d$d d!ZdS)%) annotations) TYPE_CHECKINGAnyOptionalN)PeftType) PeftConfig) BaseTunerz dict[PeftType, type[PeftConfig]]PEFT_TYPE_TO_CONFIG_MAPPINGzdict[PeftType, type[BaseTuner]]PEFT_TYPE_TO_TUNER_MAPPING PEFT_TYPE_TO_MIXED_MODEL_MAPPINGzdict[PeftType, str]PEFT_TYPE_TO_PREFIX_MAPPING config_dictdict[str, Any]returnrcCst|ddi|S)z Returns a Peft config object from a dictionary. Args: config_dict (`Dict[str, Any]`): Dictionary containing the configuration parameters. peft_typeN)r )rrr@/home/ubuntu/.local/lib/python3.10/site-packages/peft/mapping.pyget_peft_config$srdefaultF peft_configmodeltorch.nn.Module adapter_namestrlow_cpu_mem_usagebool state_dict!Optional[dict[str, torch.Tensor]]cCsV|js|jr td|jtvrtd|jdt|j}||||||d}|jS)ax Create PEFT layers and inject them into the model in-place. Currently the API does not support prompt learning methods and adaption prompt. This function is similar to [`get_peft_model`] but it does not return a [`PeftModel`] instance. Instead, it returns the original, mutated instance of the passed model. Args: peft_config (`PeftConfig`): Configuration object containing the parameters of the PEFT model. model (`torch.nn.Module`): The input model where the adapter will be injected. adapter_name (`str`, `optional`, defaults to `"default"`): The name of the adapter to be injected, if not provided, the default adapter name is used ("default"). low_cpu_mem_usage (`bool`, `optional`, defaults to `False`): Create empty adapter weights on meta device. Useful to speed up the loading process. state_dict (`dict`, *optional*, defaults to `None`) If a `state_dict` is passed here, the adapters will be injected based on the entries of the state_dict. This can be useful when the exact `target_modules` of the PEFT method is unknown, for instance because the checkpoint was created without meta data. Note that the values from the `state_dict` are not used, only the keys are used to determine the correct layers that should be adapted. zN`create_and_replace` does not support prompt learning and adaption prompt yet.z+`inject_adapter_in_model` does not support z" yet. Please use `get_peft_model`.)rrr)is_prompt_learningis_adaption_prompt ValueErrorrr keysr)rrrrr tuner_cls peft_modelrrrinject_adapter_in_model/s    r%)rrrr)rFN) rrrrrrrrrrrr) __future__rtypingrrrtorchutilsrconfigrtuners.tuners_utilsr r __annotations__r r r rr%rrrrs