>:j%SSKJr SSKrSSKrSSKrSSKrSSKJr SSKJ r J r SSK r SSK J r \R"\5r/SQrS\S'\R&SS j5rSS jrSS jrSS jrSS jrSSjrg)) annotationsN) Generator)PackageNotFoundErrormetadata)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_HINTSc#*^# Sv g![[4avn[U5R5m[HMup[ U4SjU55(dM![ U5"[U5R5SU35Ue eSnAff=f7f)zaRe-raise ImportError/AttributeError with an install hint when a multimodal dependency is missing.Nc3,># UH oT;v M g7f)N).0kwmsgs c/home/mande/repo/quber/.venv/lib/python3.13/site-packages/sentence_transformers/util/environment.py -suggest_extra_on_exception..#s0x9xsz- To install the required dependencies, run: ) ImportErrorAttributeErrorstrlowerranytypestrip)ekeywordshintrs @rsuggest_extra_on_exceptionr$s~  (!flln5NH0x0001gQ 00`ae`fghnoo6  s*B BB 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)torchcuda is_availableosenvironint distributedis_initialized device_countget_rankbackendsr'rhasattrr) importlibutil find_spechabana_frameworks.torch.hpur+) local_rankhthpus rget_device_namer?(s zz  2:: %RZZ 56J    - - / /EJJ4K4K4MPUPaPaPjPjPl4l**335JJzl##    ( ( * *  ! !   599#9#9#;#;  ! !"5 6 B33      c[U5nUSnUSU:H=(a USL=(a X;$![a gf=f)z: Checks if a package is available from the correct owner. z Home-pageNameNF)rr) package_nameownermeta home_pages rcheck_package_availabilityrGHsN %% F||+\ 0E\%J\\ s -0 ==c[SS5$)zB Returns True if the Huggingface accelerate library is available. accelerate huggingfacerGrr@ris_accelerate_availablerLTs &lM BBr@c[SS5$)z@ Returns True if the Huggingface datasets library is available. datasetsrJrKrr@ris_datasets_availablerO[s &j- @@r@c8[5=(a [5$)z Returns True if we have the required dependencies for training Sentence Transformers models, i.e. Huggingface datasets and Huggingface accelerate. )rLrOrr@ris_training_availablerQbs # $ @)>)@@r@)returnzGenerator[None, None, None])rRr)rCrrDrrRbool)rRrS) __future__r contextlibr9loggingr0collections.abcrimportlib.metadatarrr- transformersr getLogger__name__loggerr__annotations__contextmanagerr$r?rGrLrOrQrr@rr_sw" %= /   8 $>:   @ CAAr@