BNjUddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl m Z ejeZgdZded<ej&dd Zdd Zdd Zdd Zdd ZddZddZy)) 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/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/sentence_transformers/util/environment.py z-suggest_extra_on_exception..#s0x29xsz- To install the required dependencies, run: ) ImportErrorAttributeErrorstrlowerranytypestrip)ekeywordshintrs @rsuggest_extra_on_exceptionr#s  (!flln5NHd0x00d1gQ 00`ae`fghnoo6  s*B  B B9B0BBB c~tjjxrtjjS)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_initializedrris_dist_initializedr*(s-    ) ) + R0A0A0P0P0RRr)ctjjrdtjvrt tjd}nht r\tjjtjjkDrtjj}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)r%cudar'osenvironintr* device_countr&get_rankbackendsr-rhasattrr/ importlibutil find_spechabana_frameworks.torch.hpur1) local_rankhthpus rget_device_namerA3s zz 2:: %RZZ 56J "uzz'>'>'@5CTCTC]C]C_'_**335JJzl##    ( ( *  !  599#9#9#;  ! !"5 6 B33     r)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_availabilityrHSsQ %% F||+\ 0E\%S\J\\ s #& 22ctddS)zJ Returns True if the Huggingface accelerate library is available. accelerate huggingfacerHrr)ris_accelerate_availablerM_s &lM BBr)ctddS)zH Returns True if the Huggingface datasets library is available. datasetsrKrLrr)ris_datasets_availablerPfs &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. )rMrPrr)ris_training_availablerRms # $ @)>)@@r))returnzGenerator[None, None, None])rSbool)rSr)rDrrErrSrT) __future__r contextlibr;loggingr4collections.abcrimportlib.metadatarrr% transformersr getLogger__name__loggerr__annotations__contextmanagerr#r*rArHrMrPrRrr)rr`s" %= /   8 $>:   S@ CAAr)