ó
    >:j¼  ã                   óH  • S SK JrJr  S SKJr  S SKJr  S SKJrJ	r	J
r
JrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;J<r<J=r=  S SK>J?r?J@r@JArA  S SKBJCrCJDrDJErE  S SKFJGrG  / SQrHg	)
é   )ÚALLOWED_COMPUTE_DTYPESÚUPCAST_DTYPES)Úmap_cache_to_layer_device_map)Úreplace_lora_weights_loftq)6ÚCONFIG_NAMEÚINCLUDE_LINEAR_LAYERS_SHORTHANDÚSAFETENSORS_WEIGHTS_NAMEÚ5TRANSFORMERS_MODELS_TO_ADALORA_TARGET_MODULES_MAPPINGÚ4TRANSFORMERS_MODELS_TO_ADAMSS_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_BOFT_TARGET_MODULES_MAPPINGÚ1TRANSFORMERS_MODELS_TO_C3A_TARGET_MODULES_MAPPINGÚ4TRANSFORMERS_MODELS_TO_DELORA_TARGET_MODULES_MAPPINGÚ7TRANSFORMERS_MODELS_TO_FOURIERFT_TARGET_MODULES_MAPPINGÚ5TRANSFORMERS_MODELS_TO_GRALORA_TARGET_MODULES_MAPPINGÚ1TRANSFORMERS_MODELS_TO_HRA_TARGET_MODULES_MAPPINGÚ6TRANSFORMERS_MODELS_TO_IA3_FEEDFORWARD_MODULES_MAPPINGÚ1TRANSFORMERS_MODELS_TO_IA3_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_LILY_TARGET_MODULES_MAPPINGÚ6TRANSFORMERS_MODELS_TO_LNTUNING_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_LOHA_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_LOKR_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_LORA_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_MISS_TARGET_MODULES_MAPPINGÚ1TRANSFORMERS_MODELS_TO_OFT_TARGET_MODULES_MAPPINGÚ4TRANSFORMERS_MODELS_TO_PEANUT_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_POLY_TARGET_MODULES_MAPPINGÚ8TRANSFORMERS_MODELS_TO_PREFIX_TUNING_POSTPROCESS_MAPPINGÚ3TRANSFORMERS_MODELS_TO_PSOFT_TARGET_MODULES_MAPPINGÚ3TRANSFORMERS_MODELS_TO_PVERA_TARGET_MODULES_MAPPINGÚ6TRANSFORMERS_MODELS_TO_RANDLORA_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_ROAD_TARGET_MODULES_MAPPINGÚ3TRANSFORMERS_MODELS_TO_SHIRA_TARGET_MODULES_MAPPINGÚ6TRANSFORMERS_MODELS_TO_TINYLORA_TARGET_MODULES_MAPPINGÚ4TRANSFORMERS_MODELS_TO_VBLORA_TARGET_MODULES_MAPPINGÚ2TRANSFORMERS_MODELS_TO_VERA_TARGET_MODULES_MAPPINGÚ4TRANSFORMERS_MODELS_TO_WAVEFT_TARGET_MODULES_MAPPINGÚWEIGHTS_NAMEÚAuxiliaryTrainingWrapperÚModulesToSaveWrapperÚTrainableTokensWrapperÚ_freeze_adapterÚ_get_batch_sizeÚ_get_input_embeddings_nameÚ_get_submodulesÚ_is_valid_matchÚ_prepare_prompt_learning_configÚ_set_adapterÚ_set_trainableÚ&bloom_model_postprocess_past_key_valueÚcast_mixed_precision_paramsÚget_gptqmodel_quant_linearÚget_quantization_configÚid_tensor_storageÚinfer_deviceÚprepare_model_for_kbit_trainingÚ set_additional_trainable_modulesÚshift_tokens_rightÚ	transpose)ÚPeftTypeÚTaskTypeÚregister_peft_method)Úget_peft_model_state_dictÚload_peft_weightsÚset_peft_model_state_dict)ÚPeftWarning)Ar   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r   r'   r(   r)   r=   rC   r>   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r@   r6   r7   r8   rA   r   r9   r?   r   r:   rB   r;   r<   N)IÚ	constantsr   r   Úintegrationsr   Úloftq_utilsr   Úotherr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   Ú
peft_typesr=   r>   r?   Úsave_and_loadr@   rA   rB   ÚwarningrC   Ú__all__© ó    ÚP/home/mande/repo/quber/.venv/lib/python3.13/site-packages/peft/utils/__init__.pyÚ<module>rO      sq   ð÷ =Ý 7Ý 3÷7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷ 7÷p AÑ @ß bÑ bÝ  òB�rM   