o
    Gi                     @   s  d dl Z d dlmZ ddlmZ ddlmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZ ddlmZ dd	lmZ dd
l m!Z!m"Z"m#Z#m$Z$ ddl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ ddl,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZu ddlvmwZwmxZxmyZymzZz ddl{m|Z| ddl}m~Z~ ddlmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZmZ ddlmZ ddlmZmZmZmZmZmZmZ ddlmZmZ e|eZdd ZdS )    N)version   )__version__   )CONFIG_NAME#DEFAULT_HF_PARALLEL_LOADING_WORKERSDEPRECATED_REVISION_ARGSDIFFUSERS_DYNAMIC_MODULE_NAMEDIFFUSERS_LOAD_ID_FIELDSFLAX_WEIGHTS_NAMEGGUF_FILE_EXTENSIONHF_ENABLE_PARALLEL_LOADINGHF_MODULES_CACHEHUGGINGFACE_CO_RESOLVE_ENDPOINTMIN_PEFT_VERSIONONNX_EXTERNAL_WEIGHTS_NAMEONNX_WEIGHTS_NAMESAFE_WEIGHTS_INDEX_NAMESAFETENSORS_FILE_EXTENSIONSAFETENSORS_WEIGHTS_NAMEUSE_PEFT_BACKENDWEIGHTS_INDEX_NAMEWEIGHTS_NAME)_maybe_remap_transformers_class	deprecate)replace_example_docstring)get_class_from_dynamic_module)export_to_gifexport_to_objexport_to_plyexport_to_video)PushToHubMixin_add_variant_get_checkpoint_shard_files_get_model_fileextract_commit_hashhttp_user_agent)IBACKENDS_MAPPINGDIFFUSERS_SLOW_IMPORTENV_VARS_TRUE_AND_AUTO_VALUESENV_VARS_TRUE_VALUESUSE_JAXUSE_TF	USE_TORCHDummyObjectOptionalDependencyNotAvailable_LazyModuleget_objects_from_moduleis_accelerate_availableis_accelerate_versionis_aiter_availableis_aiter_versionis_av_availableis_better_profanity_availableis_bitsandbytes_availableis_bitsandbytes_versionis_bs4_availableis_cosmos_guardrail_availableis_flash_attn_3_availableis_flash_attn_availableis_flash_attn_versionis_flax_availableis_ftfy_availableis_gguf_availableis_gguf_versionis_google_colabis_hf_hub_versionis_hpu_availableis_inflect_available is_invisible_watermark_availableis_kernels_availableis_kernels_versionis_kornia_availableis_librosa_availableis_matplotlib_availableis_nltk_availableis_note_seq_availableis_nvidia_modelopt_availableis_nvidia_modelopt_versionis_onnx_availableis_opencv_availableis_optimum_quanto_availableis_optimum_quanto_versionis_peft_availableis_peft_versionis_pytorch_retinaface_availableis_safetensors_availableis_sageattention_availableis_sageattention_versionis_scipy_availableis_sentencepiece_availableis_tensorboard_availableis_timm_availableis_torch_availableis_torch_mlu_availableis_torch_npu_availableis_torch_versionis_torch_xla_availableis_torch_xla_versionis_torchao_availableis_torchao_versionis_torchsde_availableis_torchvision_availableis_transformers_availableis_transformers_versionis_unidecode_availableis_wandb_availableis_xformers_availableis_xformers_versionrequires_backends)get_module_from_nameget_submodule_by_name
load_image
load_video)
get_logger)
BaseOutput)
apply_lora_scalecheck_peft_versiondelete_adapter_layersget_adapter_nameget_peft_kwargsrecurse_remove_peft_layersscale_lora_layersset_adapter_layers!set_weights_and_activate_adaptersunscale_lora_layers)PIL_INTERPOLATIONmake_image_gridnumpy_to_pil	pt_to_pil)remote_decode)convert_all_state_dict_to_peft.convert_sai_sd_control_lora_state_dict_to_peftconvert_state_dict_to_diffusersconvert_state_dict_to_kohyaconvert_state_dict_to_peftconvert_unet_state_dict_to_peftstate_dict_all_zero)_get_detailed_type_is_valid_typec                 C   sJ   t tt | k r#d| v rd}nd|  d}|dt d7 }t|d S )NdevzThis example requires a source install from HuggingFace diffusers (see `https://huggingface.co/docs/diffusers/installation#install-from-source`),z+This example requires a minimum version of ,z but the version found is z.
)r   parser   ImportError)min_versionerror_message r   L/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/utils/__init__.pycheck_min_version   s   r   )os	packagingr    r   	constantsr   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   deprecation_utilsr   r   	doc_utilsr   dynamic_modules_utilsr   export_utilsr   r   r   r    	hub_utilsr!   r"   r#   r$   r%   r&   import_utilsr'   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   loading_utilsrp   rq   rr   rs   loggingrt   outputsru   
peft_utilsrv   rw   rx   ry   rz   r{   r|   r}   r~   r   	pil_utilsr   r   r   r   remote_utilsr   state_dict_utilsr   r   r   r   r   r   r   typing_utilsr   r   __name__loggerr   r   r   r   r   <module>   s*   T . K0$	