o p’×i\rã@sÐ dZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZmZd d lmZejd krPddlZnddlmZe e¡Zhd £Z e  !dh¡Z"ej# $dd¡ %¡Z&ej# $dd¡ %¡Z'ej# $dd¡ %¡Z(ej# $dd¡ %¡Z)ej# $dd¡ %¡Z*e*e vZ*ej+ej,ej-ej.ej/ej0dœZ1dZ2e'e"vr×e&e vr×ej3 4d¡duZ5e5rÖze d¡Z2e 6de2›d¡Wnej7yÕdZ5Yn wne 6d¡dZ5ej3 4d¡duZ8e8rze d¡Z9e 6de9›d¡Wn e:ydZ8Ynwej3 4d¡duZ;e;r.ze d¡Ze(e"vrrej3 4d ¡duoGej3 4d!¡duZ?e?rqze d ¡Z=e d!¡Z>e 6d"e=›d#e>›d¡Wnej7ypdZ?YnwndZ?e)e"vr£ej3 4d$¡duZ@e@r¢ze d$¡ZAe 6d%eA›d¡Wnej7y¡dZ@Yn wne 6d&¡dZ@ej3 4d'¡duZBze d'¡ZCe Dd(eC›¡Wn ej7yÎdZBYnwej3 4d)¡duZEze d)¡ZFe Dd*eF›¡Wn ej7yódZEYnwej3 4d+¡duZGze d+¡ZHe Dd,eH›¡Wn ej7ydZGYnwdZIej3 4d-¡duZJeJrRd.ZKdZIeKD]ZLze eL¡ZIWn ej7yBYq,weIduZJeJrRe Dd/eI›¡z.d0ZKdZMeKD]ZLze eL¡ZMWn ej7yoYqYweMduZNeNre Dd1eM›¡Wn ej7ydZNYnwej3 4d2¡duZOze d2¡ZPe Dd3eP›¡Wn ej7y²dZOYnwej3 4d4¡duZQze d4¡ZRe Dd5eR›¡Wn ej7y×dZQYnwej3 4d6¡duZSze d6¡ZTe Dd7eT›¡Wn ej7yüdZSYnwej3 4d8¡duZUz&e d8¡ZVe5r"e d¡Z2e e2¡e d9¡kr"eWd:ƒ‚e Dd;eV›¡Wn ej7y8dZUYnwej3 4d<¡duZXze d<¡ZYe Dd=eY›¡Wn ej7y]dZXYnwej3 4d>¡duZZze d>¡Z[e Dd?e[›¡Wn ej7y‚dZZYnwej3 4d@¡duZ\ze d@¡Z]e DdAe]›¡Wn ej7y§dZ\Ynwej3 4dB¡Z^ze dB¡Z_e DdCe_›¡Wn ej7yÊdZ^Ynwej3 4dD¡Z`ze dD¡Zae DdEea›¡Wn ej7yídZ`Ynwej3 4dF¡duZbze dF¡Zce DdGec›¡Wn ej7ydZbYnwej3 4dH¡duZdze dI¡Zee DdGee›¡Wn ej7y7dZdYnwej3 4dJ¡duZfze dJ¡Zge DdKeg›¡Wn ej7y\dZfYnwej3 4dL¡duZhze dM¡Zie DdNei›¡Wn ej7ydZhYnwej3 4dO¡duZjze dO¡Zke DdPek›¡Wn ej7y¦dZjYnwej3 4dQ¡duZlze dQ¡Zme DdRem›¡Wn ej7yËdZlYnwej3 4dS¡duZnze dS¡Zoe 6dTeo›¡Wn ej7yðdZnYnwej3 4dU¡duZpze dU¡Zqe DdVeq›¡Wn ej7ydZpYnwej3 4dW¡duZrerr?ze dW¡Zse 6dXes›d¡Wn ej7y>dZrYnwdYdZ„Ztej3 4d[¡duZuze d[¡Zve Dd\ev›¡Wn ej7ygdZuYnwd]ejwvpwexd^d_„ej#DƒƒZyd`da„Zzdbdc„Z{ddde„Z|dfdg„Z}dhdi„Z~djdk„Zdldm„Z€dndo„Zdpdq„Z‚drds„Zƒdtdu„Z„dvdw„Z…dxdy„Z†dzd{„Z‡d|d}„Zˆd~d„Z‰d€d„ZŠd‚dƒ„Z‹d„d…„ZŒd†d‡„Zdˆd‰„ZŽdŠd‹„ZdŒd„ZdŽd„Z‘dd‘„Z’d’d“„Z“d”d•„Z”d–d—„Z•d˜d™„Z–dšZ—d›Z˜dœZ™dZšdžZ›dŸZœd Zd¡Zžd¢ZŸd£Z d¤Z¡d¥Z¢d¦Z£d§Z¤d¨Z¥d©Z¦dªZ§d«Z¨d¬Z©d­Zªd®Z«d¯Z¬edHee¥ffd!e}e—ffd)ee˜ffd°eešffd±e‚e›ffd2eƒeœffdeze™ffd'e~ežffd+e€eŸffd4e„effdeˆe¡ffd@e‰e¢ffdBeŠe£ffdDe‹e¤ffdFeŒe¦ffdJeŽe§ffd²ee¨ffdOee©ffd$e“eªffd[e”e¬ffdSe–e«ffgƒZ­d³d´„Z®Gdµd¶„d¶e¯ƒZ°d·ee±efd¸e±d¹e±fdºd»„Z²d¸e±d¼e±fd½d¾„Z³d¸e±d¼e±fd¿dÀ„Z´d¸e±d¼e±fdÁd„Zµd¸e±d¼e±fdÃdÄ„Z¶d¸e±d¼e±fdÅdÆ„Z·dÇdÈ„Z¸GdÉdÊ„dÊe¹ƒZºGdËdÌ„dÌe ƒZ»dS)ÍzD Import utilities: Utilities related to imports and our lazy inits. éN)Ú OrderedDict)Úchain)Ú ModuleType)ÚAnyÚUnion)Úis_jinja_available)Úversion)ÚVersionÚparseé)Úlogging)éé>Ú1ÚONÚYESÚTRUEÚAUTOÚUSE_TFÚ USE_TORCHÚUSE_FLAXÚUSE_SAFETENSORSÚDIFFUSERS_SLOW_IMPORTÚFALSE)ú>z>=z==z!=z<=ú= 1.12z'Successfully imported xformers version Ú k_diffusionz*Successfully imported k-diffusion version Únote_seqz'Successfully imported note-seq version Úwandbz$Successfully imported wandb version Ú tensorboardz*Successfully imported tensorboard version Úcompelz%Successfully imported compel version Úftfyz#Successfully imported ftfy version Úbs4Úbeautifulsoup4Útorchsdez'Successfully imported torchsde version Ú imwatermarkzinvisible-watermarkz2Successfully imported invisible-watermark version Úpeftz#Successfully imported peft version Ú torchvisionz*Successfully imported torchvision version Ú sentencepiecez,Successfully imported sentencepiece version Ú matplotlibz)Successfully imported matplotlib version Útimmz Timm version cCótS©N)Ú_timm_available©r>r>úZ/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/utils/import_utils.pyÚis_timm_available@ór@Ú bitsandbytesz+Successfully imported bitsandbytes version z google.colabccs|]}| d¡VqdS)ÚCOLAB_N)Ú startswith)Ú.0Úkr>r>r?Ú Ks€rGcCr;r<)Ú_torch_availabler>r>r>r?Úis_torch_availableNrArIcCr;r<)Ú_torch_xla_availabler>r>r>r?Úis_torch_xla_availableRrArKcCr;r<)Ú_torch_npu_availabler>r>r>r?Úis_torch_npu_availableVrArMcCr;r<)Ú_flax_availabler>r>r>r?Úis_flax_availableZrArOcCr;r<)Ú_transformers_availabler>r>r>r?Úis_transformers_available^rArQcCr;r<)Ú_inflect_availabler>r>r>r?Úis_inflect_availablebrArScCr;r<)Ú_unidecode_availabler>r>r>r?Úis_unidecode_availablefrArUcCr;r<)Ú_onnx_availabler>r>r>r?Úis_onnx_availablejrArWcCr;r<)Ú_opencv_availabler>r>r>r?Úis_opencv_availablenrArYcCr;r<)Ú_scipy_availabler>r>r>r?Úis_scipy_availablerrAr[cCr;r<)Ú_librosa_availabler>r>r>r?Úis_librosa_availablevrAr]cCr;r<)Ú_xformers_availabler>r>r>r?Úis_xformers_availablezrAr_cCr;r<)Ú_accelerate_availabler>r>r>r?Úis_accelerate_available~rAracCr;r<)Ú_k_diffusion_availabler>r>r>r?Úis_k_diffusion_available‚rArccCr;r<)Ú_note_seq_availabler>r>r>r?Úis_note_seq_available†rArecCr;r<)Ú_wandb_availabler>r>r>r?Úis_wandb_availableŠrArgcCr;r<)Ú_tensorboard_availabler>r>r>r?Úis_tensorboard_availableŽrAricCr;r<)Ú_compel_availabler>r>r>r?Úis_compel_available’rArkcCr;r<)Ú_ftfy_availabler>r>r>r?Úis_ftfy_available–rArmcCr;r<)Ú_bs4_availabler>r>r>r?Úis_bs4_availablešrArocCr;r<)Ú_torchsde_availabler>r>r>r?Úis_torchsde_availabležrArqcCr;r<)Ú_invisible_watermark_availabler>r>r>r?Ú is_invisible_watermark_available¢rArscCr;r<)Ú_peft_availabler>r>r>r?Úis_peft_available¦rArucCr;r<)Ú_torchvision_availabler>r>r>r?Úis_torchvision_availableªrArwcCr;r<)Ú_matplotlib_availabler>r>r>r?Úis_matplotlib_available®rArycCr;r<)Ú_safetensors_availabler>r>r>r?Úis_safetensors_available²rAr{cCr;r<)Ú_bitsandbytes_availabler>r>r>r?Úis_bitsandbytes_available¶rAr}cCr;r<)Ú_is_google_colabr>r>r>r?Úis_google_colabºrArcCr;r<)Ú_sentencepiece_availabler>r>r>r?Úis_sentencepiece_available¾rArzÍ {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 k-diffusion library but it was not found in your environment. You can install it with pip: `pip install k-diffusion` 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` ÚonnxÚopencvÚinvisible_watermarkcsªt|ttfƒs |g}t|dƒr|jn|jj‰dd„|Dƒ}‡fdd„|Dƒ}|r/td |¡ƒ‚ˆdvr@tdd ƒr@td ˆ›d ƒ‚ˆd vrQtdd ƒrStdˆ›d ƒ‚dSdS)NÚ__name__css|]}t|VqdSr<)ÚBACKENDS_MAPPING)rEÚbackendr>r>r?rGcs€z$requires_backends..cs g|] \}}|ƒs| ˆ¡‘qSr>)Úformat)rEÚ availableÚmsg©Únamer>r?Ú ds z%requires_backends..Ú)Ú%VersatileDiffusionTextToImagePipelineÚVersatileDiffusionPipelineÚ$VersatileDiffusionDualGuidedPipelineÚ%StableDiffusionImageVariationPipelineÚUnCLIPPipelinerz4.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 ) Ú isinstanceÚlistÚtupleÚhasattrr…Ú __class__Ú ImportErrorÚjoinÚis_transformers_version)ÚobjÚbackendsÚchecksÚfailedr>r‹r?Úrequires_backends^s(ú ÿ ÿ ÿýr¢cs eZdZdZ‡fdd„Z‡ZS)Ú 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| d¡r|dvrtƒ ||¡St||jƒdS)NÚ_)Ú_load_connected_pipesÚ_is_onnx)rDÚsuperÚ __getattr__r¢Ú _backends)ÚclsÚkey©ršr>r?r¨ƒszDummyObject.__getattr__)r…Ú __module__Ú __qualname__Ú__doc__r¨Ú __classcell__r>r>r¬r?r£}sr£Úlibrary_or_versionÚ operationÚrequirement_versioncCsV|t ¡vrtdtt ¡ƒ›d|›ƒ‚t|}t|tƒr$tt |¡ƒ}||t|ƒƒS)aœ Args: Compares a library version to some requirement using a given operation. 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_FUNCÚkeysÚ ValueErrorr—r–Ústrr Úimportlib_metadatar)r±r²r³r>r>r?Úcompare_versionsŠs  r¹rcCstttƒ||ƒS)a Args: Compares the current PyTorch version to a given reference with an operation. operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A string version of PyTorch )r¹r Ú_torch_version©r²rr>r>r?Úis_torch_versionžs r¼cCótsdStttƒ||ƒS)a Args: Compares the current Transformers 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)rPr¹r Ú_transformers_versionr»r>r>r?rªó rcCr½)a Args: Compares the current Accelerate 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)r`r¹r Ú_accelerate_versionr»r>r>r?Úis_accelerate_version¸r¿rÁcCstsdStttƒ||ƒS)zú Args: Compares the current PEFT 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)Ú _peft_versionr¹r r»r>r>r?Úis_peft_versionÆr¿rÃcCr½)a Args: Compares the current k-diffusion 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)rbr¹r Ú_k_diffusion_versionr»r>r>r?Úis_k_diffusion_versionÔr¿rÅcCs0i}t|ƒD]}| d¡rqt||ƒ||<q|S)a Args: Returns a dict of object names and values in a module, while skipping private/internal objects module (ModuleType): Module to extract the objects from. Returns: dict: Dictionary of object names and corresponding values r¤)ÚdirrDÚgetattr)ÚmoduleÚobjectsrŒr>r>r?Úget_objects_from_moduleâs   rÊc@seZdZdZdS)ÚOptionalDependencyNotAvailablez^An error indicating that an optional dependency of Diffusers was not found in the environment.N)r…r­r®r¯r>r>r>r?rËösrËcsVeZdZdZd‡fdd„ Z‡fdd„Zdedefd d „Zd efd d „Z dd„Z ‡Z S)Ú _LazyModulezu Module class that surfaces all objects but only performs associated imports when the objects are requested. Nc s¢tƒ |¡t| ¡ƒ|_i|_| ¡D]\}}|D]}||j|<qqt| ¡ƒtt|  ¡Žƒ|_ ||_ ||_ t j |¡g|_|durFin||_||_||_dSr<)r§Ú__init__ÚsetrµÚ_modulesÚ_class_to_moduleÚitemsr—rÚvaluesÚ__all__Ú__file__Ú__spec__ÚosÚpathÚdirnameÚ__path__Ú_objectsÚ_nameÚ_import_structure) ÚselfrŒÚ module_fileÚimport_structureÚ module_specÚ extra_objectsr«rÒÚvaluer¬r>r?rÍs  ÿ z_LazyModule.__init__cs,tƒ ¡}|jD] }||vr| |¡q|Sr<)r§Ú__dir__rÓÚappend)rÝÚresultÚattrr¬r>r?rãs   €z_LazyModule.__dir__rŒÚreturncCsz||jvr |j|S||jvr| |¡}n ||j ¡vr*| |j|¡}t||ƒ}n td|j›d|›ƒ‚t|||ƒ|S)Nzmodule z has no attribute ) rÚrÏÚ _get_modulerÐrµrÇÚAttributeErrorr…Úsetattr)rÝrŒrârÈr>r>r?r¨s      z_LazyModule.__getattr__Ú module_namec CsLz t d||j¡WSty%}ztd|j›d|›d|›ƒ|‚d}~ww)NÚ.zFailed to import z@ because of the following error (look up to see its traceback): )Ú importlibÚ import_moduler…Ú ExceptionÚ RuntimeError)rÝrëÚer>r>r?rè)sÿÿý€ÿz_LazyModule._get_modulecCs|j|j|j|jffSr<)ršrÛrÔrÜ)rÝr>r>r?Ú __reduce__2sz_LazyModule.__reduce__)NN) r…r­r®r¯rÍrãr·rr¨rèròr°r>r>r¬r?rÌús   rÌ)¼r¯Úimportlib.utilríÚoperatorÚoprÖÚsysÚ collectionsrÚ itertoolsrÚtypesrÚtypingrrÚhuggingface_hub.utilsrÚ packagingrÚpackaging.versionr r rŽr Ú version_infor¸Úimportlib.metadataÚmetadataÚ get_loggerr…ÚloggerÚENV_VARS_TRUE_VALUESÚunionÚENV_VARS_TRUE_AND_AUTO_VALUESÚenvironÚgetÚupperrrÚUSE_JAXrrÚgtÚgeÚeqÚneÚleÚltr´rºÚutilÚ find_specrHÚinfoÚPackageNotFoundErrorrJÚ_torch_xla_versionr›rLÚ_torch_npu_versionÚ _jax_versionÚ _flax_versionrNrzÚ_safetensors_versionrPr¾ÚdebugrRÚ_inflect_versionrTÚ_unidecode_versionÚ_onnxruntime_versionrVÚ candidatesÚpkgÚ_opencv_versionrXrZÚ_scipy_versionr\Ú_librosa_versionr`rÀr^Ú_xformers_versionr¶rbrÄrdÚ_note_seq_versionrfÚ_wandb_versionrhÚ_tensorboard_versionrjÚ_compel_versionrlÚ _ftfy_versionrnÚ _bs4_versionrpÚ_torchsde_versionrrÚ_invisible_watermark_versionrtrÂrvÚ_torchvision_versionr€Ú_sentencepiece_versionrxÚ_matplotlib_versionr=Ú _timm_versionr@r|Ú_bitsandbytes_versionÚmodulesÚanyr~rIrKrMrOrQrSrUrWrYr[r]r_rarcrergrirkrmrorqrsrurwryr{r}rrÚFLAX_IMPORT_ERRORÚINFLECT_IMPORT_ERRORÚPYTORCH_IMPORT_ERRORÚONNX_IMPORT_ERRORÚOPENCV_IMPORT_ERRORÚSCIPY_IMPORT_ERRORÚLIBROSA_IMPORT_ERRORÚTRANSFORMERS_IMPORT_ERRORÚUNIDECODE_IMPORT_ERRORÚK_DIFFUSION_IMPORT_ERRORÚNOTE_SEQ_IMPORT_ERRORÚWANDB_IMPORT_ERRORÚTENSORBOARD_IMPORT_ERRORÚCOMPEL_IMPORT_ERRORÚBS4_IMPORT_ERRORÚFTFY_IMPORT_ERRORÚTORCHSDE_IMPORT_ERRORÚ INVISIBLE_WATERMARK_IMPORT_ERRORÚPEFT_IMPORT_ERRORÚSAFETENSORS_IMPORT_ERRORÚSENTENCEPIECE_IMPORT_ERRORÚBITSANDBYTES_IMPORT_ERRORr†r¢Útyper£r·r¹r¼rrÁrÃrÅrÊÚ BaseExceptionrËrÌr>r>r>r?ÚsÒ             ÿü  ÿ ÿ "  ÿû  ÿü  ÿ ÿ ÿ  ÿ ÿ€ÿ ÿ ÿ ÿ  ÿ ÿ ÿ ÿ  ÿ  ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ                       êÿ