:j{$ ddlZddlmZddlmZmZddlZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZ dd e ezezd edzd eezdzd e fdZe ezezZ dded edzd eezdzd e fdZeezZeezezZ dded edzd eezdzd edzfdZeezezZ dded edzd eezdzd edzfdZded efdZddedzfdZ y)N)Path)DictMapping) Calibration)Model)Pipeline)PLDA)BaseWaveformTransform) from_dictpipelinetoken cache_dirreturnct|tr|}nt|trtj|||}nt|trWd|vr:|j d||j d|tjd i|}n1tj|||}nt dt|d|td|d|S) Nr r checkpointr rUnsupported type (z1) for loading pipeline: expected `str` or `dict`.zCould not load pipeline: .) isinstancerstrfrom_pretraineddict setdefault TypeErrortype ValueError)r r r _pipelines z/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pyannote/audio/pipelines/utils/getter.py get_pipeliner %s (H% Hc ",,XUiX Hd # 8 #    /    Y 7 00<8>> model = get_model("hbredin/VoiceActivityDetection-PyanNet-DIHARD") >>> model = get_model("/path/to/checkpoint.ckpt") >>> model = get_model({"checkpoint": "hbredin/VoiceActivityDetection-PyanNet-DIHARD", ... "map_location": torch.device("cuda")}) See also -------- pyannote.audio.core.model.Model.from_pretrained F)r rstrictr rrz.) for loading model: expected `str` or `dict`.r) rrrrrrrreval)r"r r_models r get_modelr'JsJ% E3 &&    E E7 # %( i0%%.. e .( )   JJL Lr! calibrationcDt|tr|}|St|trtj|||}|St|tr;|j d||j d|tjdi|}|St dt|d)aDLoad pretrained calibration Parameters ---------- calibration : Calibration, str, or dict When `Calibration`, returns `calibration` as is. When `str`, assumes that this is either the path to a checkpoint or the name of a pretrained calibration on Huggingface.co and loads with `Calibration.from_pretrained(calibration)`. When `dict`, loads with `Calibration.from_pretrained(**calibration)`. token : str or bool, optional Huggingface token to be used for downloading from Huggingface hub. cache_dir: Path or str, optional Path to the folder where files downloaded from Huggingface hub are stored. Returns ------- calibration : Calibration Calibration. See also -------- pyannote.audio.core.calibration.Calibration.from_pretrained rr rrz4) for loading calibration: expected `str` or `dict`.r)rrrrrrrr)r(r rloaded_calibrations rget_calibrationr+s:+{+(( % K %(88  "  K &w.{I6(88G;G   k!2 34( )  r!pldacDt|tr|}|St|trtj|||}|St|tr;|j d||j d|tjdi|}|St dt|d)aLoad pretrained calibration Parameters ---------- plda : PLDA, str, or dict When `PLDA`, returns `plda` as is. When `str`, assumes that this is either the path to a checkpoint or the name of a pretrained PLDA on Huggingface.co and loads with `PLDA.from_pretrained(PLDA)`. When `dict`, loads with `PLDA.from_pretrained(**plda)`. token : str or bool, optional Huggingface token to be used for downloading from Huggingface hub. cache_dir: Path or str, optional Path to the folder where files downloaded from Huggingface hub are stored. Returns ------- plda : PLDA PLDA. See also -------- pyannote.audio.core.plda.PLDA.from_pretrained rr rrz-) for loading PLDA: expected `str` or `dict`.r)rr rrrrrr)r,r r loaded_pldas rget_pldar/s:$  D# **4u R   D$  '  Y/**2T2    d -( )  r! augmentationc|yt|tr|St|tr t|St dt |d)a[Load augmentation Parameter --------- augmentation : BaseWaveformTransform, or dict When `BaseWaveformTransform`, returns `augmentation` as is. When `dict`, loads with `torch_audiomentations`'s `from_config` utility function. Returns ------- augmentation : BaseWaveformTransform Augmentation. NrzH) for loading augmentation: expected `BaseWaveformTransform`, or `dict`.)rr raugmentation_from_dictrr)r0s rget_augmentationr3sW, 56,(%l33  T,/017 8 r!needscxtjj}|dk(rtjdg}||S||zSt |Dcgc]}tjd|d}}||St t |t j|Dcgc]\}}| c}}Scc}wcc}}w)a7Get devices that can be used by the pipeline Parameters ---------- needs : int, optional Number of devices needed by the pipeline Returns ------- devices : list of torch.device List of available devices. When `needs` is provided, returns that many devices. rcpuzcuda:d)torchcuda device_countdevicerangezip itertoolscycle)r4num_gpusdevicesindex_r;s r get_devicesrDszz&&(H1}<<&' =Npathlibrtypingrrr8pyannote.audio.core.calibrationrpyannote.audio.core.modelrpyannote.audio.core.pipelinerpyannote.audio.core.pldar /torch_audiomentations.core.transforms_interfacer "torch_audiomentations.utils.configr r2rrr PipelineModelr'PipelineAugmentationPipelineCalibrationr+ PipelinePLDAr/r3intrDrr!rrRs0 7+1)QR #'nt# :czD  D g% #'> > :>czD > >B-w6"C'$. #'2$2 :2czD 24 2jczD #'. . :.czD . D[ .b#7