+jQ!2UddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZddlZddlmZddlmZddlmZejeZgd Zd ed <ejd dZd!dZd"dZd#dZ d!dZ!d!dZ"d!dZ#d$dZ$d%d&dZ%dS)') annotationsN) Generator)PackageNotFoundErrormetadataversion)Any) SpecifierSet)Version)is_torch_npu_available)))pillowpil-pip install -U "sentence-transformers[image]")) read_videozQpip install -U "sentence-transformers[video]" # or [audio] for audio-only models)) torchvisionr)) torchcodeczpip install -U torchcodec)) soundfilelibrosaz-pip install -U "sentence-transformers[audio]"z!list[tuple[tuple[str, ...], str]]_DEPENDENCY_EXTRA_HINTSreturnGenerator[None, None, None]c#VK dVdS#ttf$r}t|tD]]\}}t fd|Dr=t |t|d||^d}~wwxYw)zaRe-raise ImportError/AttributeError with an install hint when a multimodal dependency is missing.Nc3 K|]}|vV dSN).0kwmsgs TC:\Projects\DSCons\.venv\Lib\site-packages\sentence_transformers\util\environment.py z-suggest_extra_on_exception..(s'0029000000z- To install the required dependencies, run: ) ImportErrorAttributeErrorstrlowerranytypestrip)ekeywordshintrs @rsuggest_extra_on_exceptionr+ s   (!ffllnn5 p pNHd0000x00000 pd1ggQggaegghhnoo p s B(BB##B(boolcztjotjS)aP Returns whether ``torch.distributed`` is available and has been initialized. The availability check must come first: some PyTorch builds (e.g. ROCm or CPU-only) report ``torch.distributed.is_available() == False`` and do not expose APIs like ``is_initialized``, so calling them directly raises ``AttributeError``. )torch distributed is_availableis_initializedrr ris_dist_initializedr2-s-   ) ) + + R0A0P0P0R0RRr r#ctjrdtjvr t tjd}nmt r]tjtj krtj }nd}d|Stj j rdStrdSttdr tjrdStjdddlmcm}|rd Sd S) aV Returns the name of the device where this module is running on. This function only supports single device or basic distributed training setups. In distributed mode for cuda device, it uses the rank to assign a specific CUDA device. Returns: str: Device name, like 'cuda:2', 'mps', 'npu', 'xpu', 'hpu', or 'cpu' LOCAL_RANKrzcuda:mpsnpuxpuhabana_frameworksNhpucpu)r.cudar0osenvironintr2 device_countr/get_rankbackendsr5r hasattrr7 importlibutil find_spechabana_frameworks.torch.hpur9) local_rankhthpus rget_device_namerI8sH z   2: % %RZ 566JJ " " uz'>'>'@'@5CTC]C]C_C_'_'_*3355JJJ#z###   ( ( * * u  ! !u   59#9#9#;#;u  ! !"5 6 6 B333333333      5 5r package_nameownerc| t|}|d}|d|ko|duo||vS#t$rYdSwxYw)zB Checks if a package is available from the correct owner. z Home-pageNameNF)rr)rJrKmeta home_pages rcheck_package_availabilityrPXsd %%% F||+\ 0E\%S\J\\ uus *- ;;c"tddS)zJ Returns True if the Huggingface accelerate library is available. accelerate huggingfacerPrr ris_accelerate_availablerUds &lM B BBr c"tddS)zH Returns True if the Huggingface datasets library is available. datasetsrSrTrr ris_datasets_availablerXks &j- @ @@r c:to tS)z Returns True if we have the required dependencies for training Sentence Transformers models, i.e. Huggingface datasets and Huggingface accelerate. )rUrXrr ris_training_availablerZrs # $ $ @)>)@)@@r str | Nonec,|dkrtjStj|dd}t t|ddx}tr|S t|S#t$rYdSwxYw)a Returns the version of an installed package, or None if the package can't be found. The ``__version__`` of an already imported module takes precedence over the distribution metadata, as the metadata of editable installs lags behind the code that actually runs. python-_ __version__N) platformpython_versionsysmodulesgetreplace isinstancegetattrr#rr)rJmodulemodule_versions rget_installed_versionrkzsx&((( [__\11#s;; < `_ version specifier, e.g. ``{"transformers": ">=5.15", "peft": ">=0.18,<0.20"}``. Instead of a specifier, a value may also be a dictionary with a ``"specifier"`` and a ``"reason"`` describing what goes wrong if the requirement isn't met. The special ``"python"`` package name is matched against the running Python version. source (str, optional): The model name or path that declared the requirements, used in the error message. Defaults to None. Raises: ImportError: If one or more requirements aren't satisfied. NzIgnoring the `requirements` of z.: expected a dictionary of version specifiers.pytorchr. specifierreasonzCould not verify the z requirement of z&: expected a version specifier string.T) prereleasesz: z==z is installedzit is not installedz- z, but z. r]"z The model z requires:zThis model requires:z5Install compatible versions with: pip install -U   )rgdictloggerwarningitemsrer#rkr containsr Exceptionappendrstripjoinr!)rlrnaliasesunmet install_specsrJ requirementis_dictrrrtinstalled_version specifier_setr(foundliness rcheck_version_requirementsrs(  lD ) )qqqqrrr'"GE!M%1%7%7%9%9AA! k{{<>> [$//8?PKOOK444[ 29A2...r)S))  NNu uuVuuu    1,?? (EEEM ,1G1GPaHbHb1c1c,    NN_9__PV__\]__ ` ` ` HHHH HYs<CC#4CCCC^s J,J JJJJ&JJQQSSTTT 8 # #  !?\!?9!?!?!? @ @ @ ,6<X2f2222BXa[`a  m LLkRURZRZ[hRiRikk l l l$))E**+++ ,,s5D99 E1#E,,E1)rr)rr,)rr#)rJr#rKr#rr,)rJr#rr[r)rlrmrnr[rro)& __future__r contextlibrCloggingr<rarccollections.abcrimportlib.metadatarrrtypingrr.packaging.specifiersr packaging.versionr transformersr getLogger__name__rzr__annotations__contextmanagerr+r2rIrPrUrXrZrkrrr rrs"""""""  %%%%%%FFFFFFFFFF ------%%%%%%//////  8 $ $>>>     SSSS@    CCCCAAAAAAAA(?,?,?,?,?,?,?,r