o
    Zh#                     @   s<  d dl mZ ddlmZmZmZmZ i ddgdg ddg d	d
g ddg dddgdg ddddgddgdg ddg dddgdg ddg d d!d"gd#d$gd%d&gd'd(giZze sne W n	 eyx   Y nw d)d*ged+< z	ed,se W n	 ey   Y nw g d-ed.< z	ed/se W n	 ey   Y nw d0ged1< er
d2d3lm	Z	 d2dl
mZmZmZmZmZ d2d	lmZmZmZmZ d2dlmZmZmZmZmZmZmZ d2dlmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z' d2d4l(m)Z) d2dl*m+Z+m,Z,m-Z- d2d5l.m/Z/m0Z0 d2d6l1m2Z2 d2dl3m4Z4m5Z5m6Z6m7Z7m8Z8 d2dl9m:Z:m;Z;m<Z<m=Z= d2d7l>m?Z? d2dl@mAZAmBZBmCZCmDZD d2d lEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZl d2d8lmmnZn d2d9lompZp d2d:lqmrZr d2d;lsmtZt z	e se W n
 ey   Y n	w d2d<lumvZvmwZw z
ed,se W n
 ey   Y nw d2d=lxmyZymzZzm{Z{ z
ed/se W n ey   Y d?S w d2d>l|m}Z} d?S d d?l~Z~eee d@ eedAe~je< d?S )B    )TYPE_CHECKING   )OptionalDependencyNotAvailable_LazyModuleis_torch_availableis_torch_greater_or_equalaqlmreplace_with_aqlm_linearawq)fuse_awq_modulespost_init_awq_exllama_modulespost_init_awq_ipex_modulesreplace_quantization_scalesreplace_with_awq_linearbitnet)	BitLinearpack_weightsreplace_with_bitnet_linearunpack_weightsbitsandbytes)dequantize_and_replaceget_keys_to_not_convertreplace_8bit_linearreplace_with_bnb_linear set_module_8bit_tensor_to_device%set_module_quantized_tensor_to_device!validate_bnb_backend_availability	deepspeed)
HfDeepSpeedConfigHfTrainerDeepSpeedConfigdeepspeed_configdeepspeed_initdeepspeed_load_checkpointdeepspeed_optim_schedis_deepspeed_availableis_deepspeed_zero3_enabledset_hf_deepspeed_configunset_hf_deepspeed_configeetqreplace_with_eetq_linear
fbgemm_fp8)FbgemmFp8LinearFbgemmFp8Llama4TextExpertsreplace_with_fbgemm_fp8_linearfinegrained_fp8	FP8Linearreplace_with_fp8_linearfsdpis_fsdp_managed_moduleggml)GGUF_CONFIG_MAPPINGGGUF_TOKENIZER_MAPPING_gguf_parse_valueload_dequant_gguf_tensor	load_ggufhiggs)HiggsLineardequantize_higgsquantize_with_higgsreplace_with_higgs_linearhqqprepare_for_hqq_linearhub_kernels)LayerRepositoryregister_kernel_mappingreplace_kernel_forward_from_hubuse_kernel_forward_from_hubintegration_utils)'INTEGRATION_TO_CALLBACKAzureMLCallbackClearMLCallbackCodeCarbonCallbackCometCallbackDagsHubCallbackDVCLiveCallbackFlyteCallbackMLflowCallbackNeptuneCallbackNeptuneMissingConfigurationSwanLabCallbackTensorBoardCallbackWandbCallback$get_available_reporting_integrations#get_reporting_integration_callbacks	hp_paramsis_azureml_availableis_clearml_availableis_codecarbon_availableis_comet_availableis_dagshub_availableis_dvclive_available is_flyte_deck_standard_availableis_flytekit_availableis_mlflow_availableis_neptune_availableis_optuna_availableis_ray_availableis_ray_tune_availableis_sigopt_availableis_swanlab_availableis_tensorboard_availableis_wandb_availablerewrite_logsrun_hp_search_optunarun_hp_search_rayrun_hp_search_sigoptrun_hp_search_wandbpeftPeftAdapterMixinquantoreplace_with_quanto_layersspqrreplace_with_spqr_linearvptqreplace_with_vptq_linear$TorchExportableModuleWithStaticCacheconvert_and_export_with_cache
executorchz2.3)shard_and_distribute_moduleSUPPORTED_TP_STYLES!translate_to_torch_parallel_styletensor_parallelz2.5make_flex_block_causal_maskflex_attention   )r	   )r)   )r/   r0   )r2   )r?   )rn   )rp   )rr   )rt   )ru   rv   )ry   rx   rz   )r|   N__file__)Zmodule_spec)typingr   utilsr   r   r   r   Z_import_structurer   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<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   rw   ru   rv   r{   ry   rx   rz   r}   r|   sys__name__globals__spec__modules r   r   Q/var/www/auris/lib/python3.10/site-packages/transformers/integrations/__init__.py<module>   s   	$%&'(/56<efghk$0	)

 