o ۷i @s dZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddlmZddlmZddlmZddlmZdd lmZmZd d lmZejd krTddlZnddlmZzea Wn e!ykda Ynwe"e#Z$hd Z%e%&dhZ'ej()dd*Z+ej()dd*Z,ej()dd*Z-ej()dd*Z.ej()dd*Z/e/e%vZ/ej0ej1ej2ej3ej4ej5dZ6dej7vpe8ddej(DZ9d de:de;deZ?n e$@d dZ>d!Z?d!ZAd!ZBe-e'vr7ejCDd"duo ejCDd#duZEeEr6zeFd"ZAeFd#ZBe$@d$eAd%eBd&WnejGy5dZEYnwndZEe.e'vrEe=d'\ZHZIne$@d(dZHd!ZJejCDd)duZKeKrd*ZLdZJeLD]ZMzeFeMZJWn ejGyuYq_weJduZKeKre$Nd+eJz.d,ZLdZOeLD]ZMzeFeMZOWn ejGyYqweOduZPePre$Nd-eOWn ejGydZPYnwejCDd.duZQzeFd/ZRe$Nd0eRWn ejGydZQYnwejCDd1duZSzeFd2ZTe$Nd3eTWn ejGy dZSYnwe=d4\ZUZVe=d5\ZWZXe=d6\ZYZZe=d7\Z[Z\e=d8\Z]Z^e=d9\Z_Z`e=d:\ZaZbe=d;\ZcZde=d<\ZeZfe=d=\ZgZhe=d>\ZiZje=d?\ZkZle=d@\ZmZne=dA\ZoZpe=dB\ZqZre=dC\ZsZte=dD\ZuZve=dE\ZwZxe=dF\ZyZze=dG\Z{Z|e=dH\Z}Z~e=dI\ZZe=dJ\ZZe=dK\ZZe=dL\ZZe=dM\ZZe=dN\ZZe=dF\ZyZze=dOdPdQ\ZZe=dR\ZZe=dS\ZZe=dT\ZZe=dU\ZZe=dV\ZZe=dW\ZZe=dX\ZZe=dYdPdQ\ZZe=dZ\ZZe=d[dPdQ\ZZe=d\\ZZd]d^Zd_d`ZdadbZdcddZdedfZdgdhZdidjZdkdlZdmdnZdodpZdqdrZdsdtZdudvZdwdxZdydzZd{d|Zd}d~ZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZddZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZe d.eeffd#eeffd:eeffdeeffdeeffdIeeffdeeffd7eeffd;eeffdJeeffdeeffd?eeffdHeeffdAeeffdeeffdBeeffd'eeffdFeeffd@eeffdGeeffdMeeffdNeeffdeeffdReeffdSeeffdTeeffgZdd܄ZGddބdeZde:eBde:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZe de:de:fddZddZGdddeZGd d d eZdS( zD Import utilities: Utilities related to imports and our lazy inits. N) OrderedDict defaultdict) lru_cache)chain) ModuleType)Any)is_jinja_available)Versionparse)logging)>1ONYESTRUEAUTOUSE_TF USE_TORCHUSE_FLAXUSE_SAFETENSORSDIFFUSERS_SLOW_IMPORTFALSE)>z>=z==z!=z<=;sr"Fpkg_name get_dist_namereturnc Cs`tj|du}d}|rtdur^ttaz7tD]0}|dp!d }dd|j p+gDdh}t dd|}|p;|D] }t| |j dqKs"z(_is_package_available..cSsd|vS)N.r namer r r!Nsz'_is_package_available..Namer z)Multiple distributions found for package z. Picked distribution: rzSuccessfully imported z version F) importlibutil find_spec _package_maprlistimportlib_metadata distributions read_textsplitfilesfilterappendmetadata Exceptionr(loggerwarningversiondebug ImportErrorPackageNotFoundError) r#r$ pkg_exists pkg_versiondist_top_level_declared_inferred_opt_names_top_level_inferredpkg_r r r!_is_package_available>sJ     rOtorchz*Disabling PyTorch because USE_TORCH is setr&jaxflaxz JAX version z, Flax version z available. safetensorsz4Disabling Safetensors because USE_SAFETENSORS is set onnxruntime) rTzonnxruntime-cannzonnxruntime-directmlort_nightly_directmlzonnxruntime-gpuort_nightly_gpuzonnxruntime-migraphxzonnxruntime-openvinozonnxruntime-qnnzonnxruntime-rocmzonnxruntime-trainingzonnxruntime-vitisaiz*Successfully imported onnxruntime version )z opencv-pythonzopencv-contrib-pythonzopencv-python-headlesszopencv-contrib-python-headlessz"Successfully imported cv2 version bs4beautifulsoup4z#Successfully imported ftfy version imwatermarkzinvisible-watermarkz2Successfully imported invisible-watermark version torch_xla torch_npu torch_mlu transformershuggingface_hubkernelsinflect unidecodenote_seqwandb tensorboardcompel sentencepiecetorchsdepeft torchvision matplotlibtimm bitsandbytesimageioftfyscipylibrosa acceleratexformersgguftorchaooptimumT)r$pytorch_retinafacebetter_profanitynltkcosmos_guardrail sageattention flash_attn flash_attn_3aiterkorniamodeloptavcCtSN)_torch_availabler r r r!is_torch_availablercCrr)_torch_xla_availabler r r r!is_torch_xla_availablerrcCrr)_torch_npu_availabler r r r!is_torch_npu_availablerrcCrr)_torch_mlu_availabler r r r!is_torch_mlu_availablerrcCrr)_flax_availabler r r r!is_flax_availablerrcCrr)_transformers_availabler r r r!is_transformers_availablerrcCrr)_inflect_availabler r r r!is_inflect_availablerrcCrr)_unidecode_availabler r r r!is_unidecode_availablerrcCrr)_onnx_availabler r r r!is_onnx_available rrcCrr)_opencv_availabler r r r!is_opencv_availablerrcCrr)_scipy_availabler r r r!is_scipy_availablerrcCrr)_librosa_availabler r r r!is_librosa_availablerrcCrr)_xformers_availabler r r r!is_xformers_availablerrcCrr)_accelerate_availabler r r r!is_accelerate_available rrcCrr)_kernels_availabler r r r!is_kernels_available$rrcCrr)_note_seq_availabler r r r!is_note_seq_available(rrcCrr)_wandb_availabler r r r!is_wandb_available,rrcCrr)_tensorboard_availabler r r r!is_tensorboard_available0rrcCrr)_compel_availabler r r r!is_compel_available4rrcCrr)_ftfy_availabler r r r!is_ftfy_available8rrcCrr)_bs4_availabler r r r!is_bs4_available<rrcCrr)_torchsde_availabler r r r!is_torchsde_available@rrcCrr)_invisible_watermark_availabler r r r! is_invisible_watermark_availableDrrcCrr)_peft_availabler r r r!is_peft_availableHrrcCrr)_torchvision_availabler r r r!is_torchvision_availableLrrcCrr)_matplotlib_availabler r r r!is_matplotlib_availablePrrcCrr)_safetensors_availabler r r r!is_safetensors_availableTrrcCrr)_bitsandbytes_availabler r r r!is_bitsandbytes_availableXrrcCrr)_is_google_colabr r r r!is_google_colab\rrcCrr)_sentencepiece_availabler r r r!is_sentencepiece_available`rrcCrr)_imageio_availabler r r r!is_imageio_availabledrrcCrr)_gguf_availabler r r r!is_gguf_availablehrrcCrr)_torchao_availabler r r r!is_torchao_availablelrrcCrr)_optimum_quanto_availabler r r r!is_optimum_quanto_availableprrcCrr)_nvidia_modelopt_availabler r r r!is_nvidia_modelopt_availabletrrcCrr)_timm_availabler r r r!is_timm_availablexrrcCrr)_pytorch_retinaface_availabler r r r!is_pytorch_retinaface_available|rrcCrr)_better_profanity_availabler r r r!is_better_profanity_availablerrcCrr)_nltk_availabler r r r!is_nltk_availablerrcCrr)_cosmos_guardrail_availabler r r r!is_cosmos_guardrail_availablerrcCstdddDS)Ncss|] }tj|VqdSr)r3r4r5)rlibr r r!r"sz#is_hpu_available..)habana_frameworkszhabana_frameworks.torch)allr r r r!is_hpu_availablesrcCrr)_sageattention_availabler r r r!is_sageattention_availablerrcCrr)_flash_attn_availabler r r r!is_flash_attn_availablerrcCrr)_flash_attn_3_availabler r r r!is_flash_attn_3_availablerrcCrr)_aiter_availabler r r r!is_aiter_availablerrcCrr)_kornia_availabler r r r!is_kornia_availablerrcCrr) _av_availabler r r r!is_av_availablerrz {0} requires the FLAX library but it was not found in your environment. Checkout the instructions on the installation page: https://github.com/google/flax and follow the ones that match your environment. z {0} requires the inflect library but it was not found in your environment. You can install it with pip: `pip install inflect` z {0} requires the PyTorch library but it was not found in your environment. Checkout the instructions on the installation page: https://pytorch.org/get-started/locally/ and follow the ones that match your environment. z {0} requires the onnxruntime library but it was not found in your environment. You can install it with pip: `pip install onnxruntime` z {0} requires the OpenCV library but it was not found in your environment. You can install it with pip: `pip install opencv-python` z{ {0} requires the scipy library but it was not found in your environment. You can install it with pip: `pip install scipy` z {0} requires the librosa library but it was not found in your environment. Checkout the instructions on the installation page: https://librosa.org/doc/latest/install.html and follow the ones that match your environment. z {0} requires the transformers library but it was not found in your environment. You can install it with pip: `pip install transformers` z {0} requires the unidecode library but it was not found in your environment. You can install it with pip: `pip install Unidecode` z {0} requires the note-seq library but it was not found in your environment. You can install it with pip: `pip install note-seq` z{ {0} requires the wandb library but it was not found in your environment. You can install it with pip: `pip install wandb` z {0} requires the tensorboard library but it was not found in your environment. You can install it with pip: `pip install tensorboard` z} {0} requires the compel library but it was not found in your environment. You can install it with pip: `pip install compel` z {0} requires the Beautiful Soup library but it was not found in your environment. You can install it with pip: `pip install beautifulsoup4`. Please note that you may need to restart your runtime after installation. a8 {0} requires the ftfy library but it was not found in your environment. Checkout the instructions on the installation section: https://github.com/rspeer/python-ftfy/tree/master#installing and follow the ones that match your environment. Please note that you may need to restart your runtime after installation. z {0} requires the torchsde library but it was not found in your environment. You can install it with pip: `pip install torchsde` z {0} requires the invisible-watermark library but it was not found in your environment. You can install it with pip: `pip install invisible-watermark>=0.2.0` zy {0} requires the peft library but it was not found in your environment. You can install it with pip: `pip install peft` z {0} requires the safetensors library but it was not found in your environment. You can install it with pip: `pip install safetensors` z {0} requires the sentencepiece library but it was not found in your environment. You can install it with pip: `pip install sentencepiece` z {0} requires the bitsandbytes library but it was not found in your environment. You can install it with pip: `pip install bitsandbytes` z {0} requires the imageio library and ffmpeg but it was not found in your environment. You can install it with pip: `pip install imageio imageio-ffmpeg` zy {0} requires the gguf library but it was not found in your environment. You can install it with pip: `pip install gguf` z {0} requires the torchao library but it was not found in your environment. You can install it with pip: `pip install torchao` z {0} requires the optimum-quanto library but it was not found in your environment. You can install it with pip: `pip install optimum-quanto` z {0} requires the pytorch_retinaface library but it was not found in your environment. You can install it with pip: `pip install pytorch_retinaface` z {0} requires the better_profanity library but it was not found in your environment. You can install it with pip: `pip install better_profanity` zy {0} requires the nltk library but it was not found in your environment. You can install it with pip: `pip install nltk` onnxopencvinvisible_watermarkquantocst|ttfs |g}t|dr|jn|jjdd|D}fdd|D}|r/td|dvr@tdd r@td d d vrQtdd rStdd dSdS)N__name__css|]}t|VqdSr)BACKENDS_MAPPING)rbackendr r r!r"msz$requires_backends..cs g|] \}}|s|qSr )format)r availablemsgr/r r! ns z%requires_backends..r')%VersatileDiffusionTextToImagePipelineVersatileDiffusionPipeline$VersatileDiffusionDualGuidedPipeline%StableDiffusionImageVariationPipelineUnCLIPPipelinerz4.25.0z9You need to install `transformers>=4.25` in order to use z/: ``` pip install --upgrade transformers ```) StableDiffusionDepth2ImgPipeline"StableDiffusionPix2PixZeroPipelinez4.26.0z9You need to install `transformers>=4.26` in order to use ) isinstancer7tuplehasattrr __class__rEjoinis_transformers_version)objbackendschecksfailedr r/r!requires_backendshs(   rcs eZdZdZfddZZS) DummyObjectz Metaclass for the dummy objects. Any class inheriting from it will return the ImportError generated by `requires_backend` each time a user tries to access any method of that class. cs0|dr|dvrt||St||jdS)NrN)_load_connected_pipes_is_onnx)rsuper __getattr__r _backends)clskeyrr r!rszDummyObject.__getattr__)r __module__ __qualname____doc__r __classcell__r r r r!rsrlibrary_or_version operationrequirement_versioncCsV|tvrtdttd|t|}t|tr$tt|}||t|S)a Compares a library version to some requirement using a given operation. Args: library_or_version (`str` or `packaging.version.Version`): A library name or a version to check. operation (`str`): A string representation of an operator, such as `">"` or `"<="`. requirement_version (`str`): The version to compare the library version against z`operation` must be one of z , received ) STR_OPERATION_TO_FUNCkeys ValueErrorr7rstrr r8rC)rrrr r r!compare_versionss  rrCcCsttt||S)a  Compares the current PyTorch version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A string version of PyTorch )rr _torch_versionrrCr r r!is_torch_versions rcCtsdSttt||S)a  Compares the current torch_xla version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A string version of torch_xla F)rrr _torch_xla_versionrr r r!is_torch_xla_version rcCr)a Compares the current Transformers version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _transformers_versionrr r r!rrrcCr)z Compares the current Kernels version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _kernels_versionrr r r!is_kernels_versionrrcCr)a Compares the current Hugging Face Hub version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)_hf_hub_availablerr _hf_hub_versionrr r r!is_hf_hub_versionrr"cCra Compares the current Accelerate version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _accelerate_versionrr r r!is_accelerate_versionrr%cCr)z Compares the current PEFT version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _peft_versionrr r r!is_peft_versionrr'cCr)a Args: Compares the current bitsandbytes version to a given reference with an operation. operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _bitsandbytes_versionrr r r!is_bitsandbytes_versions r)cCrr#)rrr _gguf_versionrr r r!is_gguf_version&rr+cCr)z Compares the current torchao version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _torchao_versionrr r r!is_torchao_version6rr-cCrr#)rrr _optimum_quanto_versionrr r r!is_optimum_quanto_versionFrr/cCr)a Compares the current Nvidia ModelOpt version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _nvidia_modelopt_versionrr r r!is_nvidia_modelopt_versionVrr1cCr)z Compares the current xformers version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _xformers_versionrr r r!is_xformers_versionfrr3cCr)a Compares the current sageattention version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _sageattention_versionrr r r!is_sageattention_versionvrr5cCr)a Compares the current flash-attention version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _flash_attn_versionrr r r!is_flash_attn_versionrr7cCr)z Compares the current aiter version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A version string F)rrr _aiter_versionrr r r!is_aiter_versionrr9cCs0i}t|D]}|drqt||||<q|S)a Returns a dict of object names and values in a module, while skipping private/internal objects Args: module (ModuleType): Module to extract the objects from. Returns: dict: Dictionary of object names and corresponding values rN)dirrgetattr)moduleobjectsr0r r r!get_objects_from_modules   r>c@seZdZdZdS)OptionalDependencyNotAvailablezh An error indicating that an optional dependency of Diffusers was not found in the environment. N)rr r r r r r r!r?sr?csVeZdZdZdfdd ZfddZdedefd d Zd efd d Z ddZ Z S) _LazyModulezu Module class that surfaces all objects but only performs associated imports when the objects are requested. Nc st|t||_i|_|D]\}}|D]}||j|<qqt|tt| |_ ||_ ||_ t j|g|_|durFin||_||_||_dSr)r__init__setr_modules_class_to_moduleitemsr7rvalues__all____file____spec__ospathdirname__path___objects_name_import_structure) selfr0 module_fileimport_structure module_spec extra_objectsrrFvaluer r r!rAs   z_LazyModule.__init__cs,t}|jD] }||vr||q|Sr)r__dir__rGr>)rQresultattrr r r!rWs   z_LazyModule.__dir__r0r%cCsz||jvr |j|S||jvr||}n ||jvr*||j|}t||}n td|jd|t||||S)Nzmodule z has no attribute ) rNrC _get_modulerDrr;AttributeErrorrsetattr)rQr0rVr<r r r!rs      z_LazyModule.__getattr__ module_namec CsLz td||jWSty%}ztd|jd|d||d}~ww)Nr.zFailed to import z@ because of the following error (look up to see its traceback): )r3 import_modulerr@ RuntimeError)rQr]er r r!rZsz_LazyModule._get_modulecCs|j|j|j|jffSr)rrOrHrP)rQr r r! __reduce__sz_LazyModule.__reduce__)NN) rr r r rArWrrrrZrar r r r r!r@s   r@)F(r importlib.utilr3r*operatoroprJsys collectionsrr functoolsrcache itertoolsrtypesrtypingrhuggingface_hub.utilsrpackaging.versionr r r'r version_infor8importlib.metadatar?packages_distributionsr6r@ get_loggerrrAENV_VARS_TRUE_VALUESunionENV_VARS_TRUE_AND_AUTO_VALUESenvirongetupperrrUSE_JAXrrgtgeeqneleltrmodulesanyrrboolrrOrrinfo _jax_version _flax_versionr4r5rrCrFr_safetensors_version_onnxruntime_versionr candidatesrMrD_opencv_versionrr _bs4_versionr_invisible_watermark_versionrrr_torch_npu_versionr_torch_mlu_versionrrr r!rrr_inflect_versionr_unidecode_versionr_note_seq_versionr_wandb_versionr_tensorboard_versionr_compel_versionr_sentencepiece_versionr_torchsde_versionrr&r_torchvision_versionr_matplotlib_versionr _timm_versionrr(r_imageio_versionr _ftfy_versionr_scipy_versionr_librosa_versionrr$rr2rr*rr,rr.r_pytorch_retinaface_versionr_better_profanity_versionr _nltk_versionr_cosmos_guardrail_versionrr4rr6r_flash_attn_3_versionrr8r_kornia_versionrr0r _av_versionrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrFLAX_IMPORT_ERRORINFLECT_IMPORT_ERRORPYTORCH_IMPORT_ERRORONNX_IMPORT_ERROROPENCV_IMPORT_ERRORSCIPY_IMPORT_ERRORLIBROSA_IMPORT_ERRORTRANSFORMERS_IMPORT_ERRORUNIDECODE_IMPORT_ERRORNOTE_SEQ_IMPORT_ERRORWANDB_IMPORT_ERRORTENSORBOARD_IMPORT_ERRORCOMPEL_IMPORT_ERRORBS4_IMPORT_ERRORFTFY_IMPORT_ERRORTORCHSDE_IMPORT_ERROR INVISIBLE_WATERMARK_IMPORT_ERRORPEFT_IMPORT_ERRORSAFETENSORS_IMPORT_ERRORSENTENCEPIECE_IMPORT_ERRORBITSANDBYTES_IMPORT_ERRORIMAGEIO_IMPORT_ERRORGGUF_IMPORT_ERRORTORCHAO_IMPORT_ERRORQUANTO_IMPORT_ERRORPYTORCH_RETINAFACE_IMPORT_ERRORBETTER_PROFANITY_IMPORT_ERRORNLTK_IMPORT_ERRORrrtyperrrrrrr"r%r'r)r+r-r/r1r3r5r7r9r> BaseExceptionr?r@r r r r!s>              "$  "                                                                         "