#iUddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl m Z ejeZgdZded<ej&dd Zdd Zdd Zdd Zdd ZddZy)) 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#K dy#ttf$ro}t|jtD]G\}}t fd|Dst |t|jd||d}~wwxYww)zaRe-raise ImportError/AttributeError with an install hint when a multimodal dependency is missing.Nc3&K|]}|v yw)N).0kwmsgs w/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/sentence_transformers/util/environment.py z-suggest_extra_on_exception..#s0290sz- To install the required dependencies, run: ) ImportErrorAttributeErrorstrlowerranytypestrip)ekeywordshintrs @rsuggest_extra_on_exceptionr#s  (!flln5 pNHd0x00d1gQ 00`ae`fghnoo p  s*B  B B9B0BBB ctjjrdtjvrt tjd}n|tj jr\tjjtj jkDrtj j}nd}d|Stjjjrytryttdrtjjrytj j#dddlmcm}|jry y ) 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)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     cj t|}|d}|d|k(xr |duxr||vS#t$rYywxYw)zB Checks if a package is available from the correct owner. z Home-pageNameNF)rr) package_nameownermeta home_pages rcheck_package_availabilityrFHsQ %% F||+\ 0E\%S\J\\ s #& 22ctddS)zJ Returns True if the Huggingface accelerate library is available. accelerate huggingfacerFrr?ris_accelerate_availablerKTs &lM BBr?ctddS)zH Returns True if the Huggingface datasets library is available. datasetsrIrJrr?ris_datasets_availablerN[s &j- @@r?c.txr tS)z Returns True if we have the required dependencies for training Sentence Transformers models, i.e. Huggingface datasets and Huggingface accelerate. )rKrNrr?ris_training_availablerPbs # $ @)>)@@r?)returnzGenerator[None, None, None])rQr)rBrrCrrQbool)rQrR) __future__r contextlibr8loggingr/collections.abcrimportlib.metadatarrr, transformersr getLogger__name__loggerr__annotations__contextmanagerr#r>rFrKrNrPrr?rr^sy" %= /   8 $>:   @ CAAr?