rh7 ddlmZddlZddlZddlZddlmZmZddlZddl m Z eje Z d dZd dZd dZd dZd d Zy) ) annotationsN)PackageNotFoundErrormetadata)is_torch_npu_availablecztjjrdtjvrt tjd}n|tj jr\tjjtj jkDrtj j}nd}d|Stjjjrytrytjjdddlmcm}|jryy ) aO 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', 'hpu', or 'cpu' LOCAL_RANKrzcuda:mpsnpuhabana_frameworksNhpucpu)torchcuda is_availableosenvironint distributedis_initialized device_countget_rankbackendsr r importlibutil find_spechabana_frameworks.torch.hpur ) local_rankhthpus y/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/sentence_transformers/util/environment.pyget_device_namer s zz 2:: %RZZ 56J    - - /EJJ4K4K4MPUPaPaPjPjPl4l**335JJzl##    ( ( *  !  ! !"5 6 B33     cZ t|}|d|k(xr||dvS#t$rYywxYw)zB Checks if a package is available from the correct owner. Namez Home-pageF)rr) package_nameownermetas rcheck_package_availabilityr',sA %F||+Jk9J0JJ s  **ctddS)zJ Returns True if the Huggingface accelerate library is available. accelerate huggingfacer'r!ris_accelerate_availabler-7s &lM BBr!ctddS)zH Returns True if the Huggingface datasets library is available. datasetsr*r+r,r!ris_datasets_availabler0>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. )r-r0r,r!ris_training_availabler2Es # $ @)>)@@r!)returnstr)r$r4r%r4r3bool)r3r5) __future__rrloggingrimportlib.metadatarrr transformersr getLogger__name__loggerr r'r-r0r2r,r!rr=sH" = /   8 $<CAAr!