o NÆÏi“ ã@shddlmZddlZddlmZddlmZddlmZeƒr'Gdd„deƒZ d ej j d e fd d „Z dS) é)ÚOptionalN)Úis_inc_available)ÚBaseTunerLayeré)ÚLinearcsTeZdZdejjdef‡fdd„ Zddede e ed dfd d „Z dd d „Z ‡Z S)Ú IncLoraLinearÚ base_layerÚ adapter_namec stƒj||fi|¤ŽdS)N)ÚsuperÚ__init__)Úselfrr Úkwargs©Ú __class__©úH/home/ubuntu/.local/lib/python3.10/site-packages/peft/tuners/lora/inc.pyr szIncLoraLinear.__init__FNÚ safe_mergeÚ adapter_namesÚreturncCótdƒ‚)a† Merge the active adapter weights into the base weights Args: safe_merge (`bool`, *optional*): If True, the merge operation will be performed in a copy of the original weights and check for NaNs before merging the weights. This is useful if you want to check if the merge operation will produce NaNs. Defaults to `False`. adapter_names (`list[str]`, *optional*): The list of adapter names that should be merged. If None, all active adapters will be merged. Defaults to `None`. z3Merging LoRA with INC layers is not yet implemented©ÚNotImplementedError)r rrrrrÚmerge(s zIncLoraLinear.mergecCr)z_ This method unmerges all merged adapter layers from the base weights. z5Unmerging LoRA from INC layers is not yet implementedr)r rrrÚunmerge7szIncLoraLinear.unmerge)FN)rN)Ú__name__Ú __module__Ú __qualname__ÚtorchÚnnÚModuleÚstrr ÚboolrÚlistrrÚ __classcell__rrrrrsþý rÚtargetr cKsNd}t|tƒr | ¡}n|}tƒr%ddlm}t||ƒr%t||fi|¤Ž}|S)Nr)Ú PatchedLinear)Ú isinstancerÚget_base_layerrÚIneural_compressor.torch.algorithms.fp8_quant._quant_common.helper_modulesr%r)r$r r Ú new_moduleÚtarget_base_layerr%rrrÚ dispatch_inc>s    r+)ÚtypingrrÚpeft.import_utilsrÚpeft.tuners.tuners_utilsrÚlayerrrrrr r+rrrrÚs