jD<SrSSKJrJr SSKrSSKrSSKrSSKrSSKrSSK r SSK r SSKrSSK r SSK r SSK r SSKrSSKJr SSKrSSKrSSKrSSKrSSKrSSKJs Jr SSKrSSKJr SSKJrJr SSK J!r! SSK"J#r#J$r$J%r% SS K&J'r'J(r(J)r) 0S S _S S _SS _SS _SS_SS _SS_SS_SS_SS_SS_SS_SS _S!S"_S#S$_S%S&_S'S(_S)S*S+.Er*S,r+S-r,"S.S/5r-g)0zZThis file contains the Separator class, to facilitate the separation of stems from audio. )metadata resourcesN)Optional)tqdm)atomic_output_pathvalidate_audio_source) Ensembler)AudioExportErrorBatchSeparationErrorInvalidAudioDataError)AUTOCASTFP32 NATIVE_FP16vocalsVocals instrumental InstrumentalinstkaraokeotherOther no_vocalsdrumsDrumsbassBassguitarGuitarpianoPiano synthesizer SynthesizerstringsStrings woodwinds WoodwindsbrassBrassz wind instz Wind Instz lead vocalsz Lead Vocalszbacking vocalszBacking Vocalsz Primary StemzSecondary Stem)z primary stemzsecondary stem).wavz.flacz.mp3z.oggz.opusz.m4az.aiffz.ac3c## [R"U5HZupnUHNnUR5R[5(dM-[R R X5v MP M\ g7fN)oswalklowerendswithSUPPORTED_AUDIO_EXTENSIONSpathjoin) directoryroot_dirsfilesfilenames ڱC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\audio_separator/separator/separator.py_iter_directory_audio_filesr99sS ggi0UH~~(()CDDggll4221s AA8*A8c\rSrSrSr\R SSSSSSSSSS SSSSS S S S SS.S SSSSSSS.SSS SS.S SSSSS.SSSSSS4Sjr\S5r \S5r /SQr Sr Sr SrS rS!rS"rS#rS$rS%rS&rS'rS(rS)rS*rS+rS,rS-rS.rS/rS9S0jrS:S1jr S:S2jr!S:S3jr"S4r#S:S5\$\%4S6jjr&S:S7jr'S8r(g); Separator@a The Separator class is designed to facilitate the separation of audio sources from a given audio file. It supports various separation architectures and models, including MDX, VR, and Demucs. The class provides functionalities to configure separation parameters, load models, and perform audio source separation. It also handles logging, normalization, and output formatting of the separated audio stems. The actual separation task is handled by one of the architecture-specific classes in the `architectures` module; this class is responsible for initialising logging, configuring hardware acceleration, loading the model, initiating the separation process and passing outputs back to the caller. Common Attributes: log_level (int): The logging level. log_formatter (logging.Formatter): The logging formatter. model_file_dir (str): The directory where model files are stored. output_dir (str): The directory where output files will be saved. output_format (str): The format of the output audio file. output_bitrate (str): The bitrate of the output audio file. amplification_threshold (float): The threshold for audio amplification. normalization_threshold (float): The threshold for audio normalization. output_single_stem (str): Option to output a single stem. invert_using_spec (bool): Flag to invert using spectrogram. sample_rate (int): The sample rate of the audio. use_soundfile (bool): Use soundfile for audio writing, can solve OOM issues. use_autocast (bool): Use PyTorch autocast when the loaded model and device support it. use_torch_compile (bool): Compile verified repeated model blocks when supported. use_native_fp16 (bool): Convert a verified model to native float16 inference when supported. MDX Architecture Specific Attributes: hop_length (int): The hop length for STFT. segment_size (int): The segment size for processing. overlap (float): The overlap between segments. batch_size (int): The batch size for processing. enable_denoise (bool): Flag to enable or disable denoising. VR Architecture Specific Attributes & Defaults: batch_size: 16 window_size: 512 aggression: 5 enable_tta: False enable_post_process: False post_process_threshold: 0.2 high_end_process: False Demucs Architecture Specific Attributes & Defaults: segment_size: "Default" shifts: 2 overlap: 0.25 segments_enabled: True MDXC Architecture Specific Attributes & Defaults: segment_size: 256 override_model_segment_size: False batch_size: None (uses inference.batch_size from the model YAML, falling back to 1) overlap: None (uses inference.num_overlap from the model YAML, falling back to 8) pitch_shift: 0 Nz/tmp/audio-separator-models/WAVg?gFiDig?) hop_length segment_sizeoverlap batch_sizeenable_denoiseig?)rC window_size aggression enable_ttaenable_post_processpost_process_thresholdhigh_end_processDefaultT)rAshiftsrBsegments_enabledr)rAoverride_model_segment_sizerCrB pitch_shiftc. U (aU(a [S5e[R"[5UlURR U5 XlX l[R"5Ul URc[R"S5UlURRUR5 URR5(d%URRUR5 U[R:a[R "S5 U(d?UR#S5R$nURR'SUSUSU35 Uc7[(R*"5nU(dURR'S 5 X@l[(R.R1S 5nU(avUUlURR'S UR235 [(R4R7UR25(d[9S UR235eO$URR'S U35 X0l[(R:"UR2SS9 [(R:"UR,SS9 XPlX`lUR<cSUlXpl US::dUS:a [S5eXl!US:dUS:a [S5eXl"U bURRGSU S35 Xl$URH(aURRGS5 [KU 5Ul&URLS::a[SURLS35eURLS:a[SURLS35eXl'Xl(UUl)UUl*Xl+Xl,UbUS::a [S5eUUl-UUl.UUl/SUl0UbGURcU5nUSUl0Uc USUl-UcUR1S 5b US Ul.URZcS!Ul-UUUUS".Ul2SUl3SUl4SUl5SUl6SUl7SUl8/Ul9SUl:SUl;S#Ul<S#Ul=SUl>U(dUR5 gg![a [S5ef=f)$zInitialize the separator.zHuse_autocast and use_native_fp16 are mutually exclusive precision modes.Nz6%(asctime)s - %(levelname)s - %(module)s - %(message)signorezaudio-separatorzSeparator version z instantiating with output_dir: z, output_format: z@Output directory not specified. Using current working directory.AUDIO_SEPARATOR_MODEL_DIRz>Using model directory from AUDIO_SEPARATOR_MODEL_DIR env var: z.The specified model directory does not exist: z5Using model directory from model_file_dir parameter: T)exist_okr=rr?zOThe normalization_threshold must be greater than 0 and less than or equal to 1.z[The amplification_threshold must be greater than or equal to 0 and less than or equal to 1.z7Single stem output requested, so only one output file (z) will be writtenzwSecondary step will be inverted using spectogram rather than waveform. This may improve quality but is slightly slower.zThe sample rate setting is z( but it must be a non-zero whole number.iPz!. Enter something less ambitious.zPThe sample rate must be a non-zero whole number. Please provide a valid integer.z%chunk_duration must be greater than 0models algorithmweightsavg_waveMDXVRDemucsMDXCF)@ ValueErrorlogging getLogger__name__loggersetLevel log_level log_formatter StreamHandler log_handler Formatter setFormatter hasHandlers addHandlerDEBUGwarningsfilterwarningsget_package_distributionversioninfor,getcwd output_direnvirongetmodel_file_dirr1existsFileNotFoundErrormakedirs output_formatoutput_bitratenormalization_thresholdamplification_thresholdoutput_single_stemdebuginvert_using_specint sample_rate use_soundfile use_autocastuse_native_fp16use_torch_compile use_directmlchunk_durationensemble_algorithmensemble_weightsensemble_preset_ensemble_preset_models_load_ensemble_presetarch_specific_params torch_devicetorch_device_cputorch_device_mpsonnx_execution_providermodel_instancemodel_filenamemodel_filenames_loaded_model_filename_loaded_model_friendly_name_loaded_model_is_uvr_vipmodel_is_uvr_vipmodel_friendly_name$setup_accelerated_inferencing_device)selfrerfrwrtr{r|r}r~rrrrrrr mdx_params vr_params demucs_params mdxc_paramsrrr info_onlyrrpackage_version env_model_dir preset_datas r8__init__Separator.__init__zs: Ogh h''1  Y'"*"002    %!(!2!23k!lD  %%d&8&89{{&&(( KK " "4#3#3 4 w}} $  # #H -";;>$"5"566'*XY]YlYlXm(noo7 KK  TUcTde f"0  D''$7 DOOd3*,    %!&D '>$ "a '+BQ+Fno o'>$ "Q &*AA*Ez{ {"4  ) KK   WXjWkk|} ~!2  ! ! KK  !X Z q";/D 1$ #>t?O?O>PPx!yzz(* #>t?O?O>PPq!rss+ +(.!2(,  %" !HII"4 0.'+$  &44_EK+6x+@D (!)*5k*B''KOOI,F,R(3I(>%  " " *&0D #-7iS`ju$v!  $ $'+$""!&*#+/((-% %#'   5 5 7i qop p qs .A"S>>Tcx[[USS55S:a[$[URS[5$)zBReturn the precision mode selected for the currently loaded model.rr?effective_precision)lengetattrrrrs r8rSeparator.effective_precisions6 wt.3 4q 8Kt**,A4HHcx[[USS55S:ag[[URSS55$)zCReturn whether regional compilation is active for the loaded model.rrr?Feffective_torch_compile)rrboolrrs r8r!Separator.effective_torch_compile"s9 wt.3 4q 8GD//1JERSSr) rY median_wavemin_wavemax_waveavg_fft median_fftmin_fftmax_fft uvr_max_spec uvr_min_spec ensemble_wavc 8[R"SS5n[R"U5nSSS5 WR S05nX;a9SR[UR555n[ SUSU35eXAnUR S /5n[U[5(a[U5S :a7[ S US [U[5(a[U535eS 35eUR SS5nXR;a[ S USUS35eUR S5n U bk[U [5(a[U 5[U5:wa>[ S US[U [5(a [U 5OSS[U5S35eURRSUSUR SU5SUR SS535 U$!,(df  GN=f![a [ S5ef=f)z Load and validate an ensemble preset from ensemble_presets.json. Returns a dict with keys: name, description, models, algorithm, weights, contributor. Raises ValueError if the preset is not found or fails validation. audio_separatorensemble_presets.jsonNzVEnsemble presets file not found. The package may be corrupted or improperly installed.presets, zUnknown ensemble preset: 'z'. Available presets: rVrMzEnsemble preset 'z&' must specify at least 2 models, got rrWrYz' has unknown algorithm: ''rXz' weights length (zN/Az) must match models count ()zLoaded ensemble preset 'z': nameu — description)r open_textjsonloadryr_rvr2sortedkeys isinstancelistrVALID_ENSEMBLE_ALGORITHMSrcrr) r preset_namef presets_datar availablepresetrVrWrXs r8rSeparator._load_ensemble_preset/s w$$%68OPTU#yy| Q ""9b1  % &"89I9+F\]f\ghi i%Hb)&$''3v;?0 =cs}EGKtLtLdghndodTUV VRSdTUV VJJ{J7 :: :0 =WXaWbbcde e**Y'  gt,,G F 0K #4[MASdnovx|d}d}TWX_T`DITJJefijpfqerrs"tuu 3K=FJJvWbDcCddijpjtjtvCEGkHjIJ K ;QP  wuv v ws'HG1H1 H;HHHc[R"SS5n[R"U5nSSS5 WR S05$!,(df  N =f![a 0s$f=f)zd List all available ensemble presets. Returns a dict mapping preset IDs to their full preset data. rrNr)rrrrryrv)rrrs r8list_ensemble_presetsSeparator.list_ensemble_presetsVsa  $$%68OPTU#yy| Q 2.. QP  I s-AA A AAA A*)A*cUR5nUR5 UR5 UR5 UR U5 g)w This method sets up the PyTorch and/or ONNX Runtime inferencing device, using GPU hardware acceleration if available. N)get_system_infocheck_ffmpeg_installedlog_onnxruntime_packages preload_onnxruntime_dependenciessetup_torch_device)r system_infos r8r.Separator.setup_accelerated_inferencing_devicecsD**,  ##% %%' --/  ,rc2[[S5(dURRS5 g[R"5 URRS5 g![ a(nURR SU35 SnAgSnAff=f)z Preload ONNX Runtime shared library dependencies when supported by the installed package. This helps pip-installed CUDA/cuDNN runtime wheels become visible to ONNX Runtime before the first CUDAExecutionProvider session is created. preload_dllszTInstalled ONNX Runtime does not provide preload_dlls(); skipping dependency preload.Nz3Preloaded ONNX Runtime shared library dependencies.zPPZ[f[n[nZooy{F{N{NzOOVWbWlWlVmn o!002 +N+;<=++ ,_,=>?rc[R"SS/SS9nUR5SnURR SU35 g ![ a3 URR S5 S[R;aeg f=f) zA This method checks if ffmpeg is installed and logs its version. ffmpegz-versionT)textrzFFmpeg installed: zCFFmpeg is not installed. Please install FFmpeg to use this package.PYTEST_CURRENT_TESTN) subprocess check_output splitlinesrcrrryerrorr,ru)rffmpeg_version_output first_lines r8r Separator.check_ffmpeg_installeds $.$;$;XzJ KK  1*> ?   KK  c d%BJJ67  sAA :BBcURS5nURS5nURS5nURS5nUb(URRSUR35 Ub(URRSUR35 Ub(URRSUR35 Ub)URRS UR35 gg) zj This method logs the ONNX Runtime package versions, including the GPU and Silicon packages if available. zonnxruntime-gpuzonnxruntime-silicon onnxruntimeonnxruntime-directmlNz1ONNX Runtime GPU package installed with version: z5ONNX Runtime Silicon package installed with version: z1ONNX Runtime CPU package installed with version: z6ONNX Runtime DirectML package installed with version: )rprcrrrq)ronnxruntime_gpu_packageonnxruntime_silicon_packageonnxruntime_cpu_packageonnxruntime_dml_packages r8r"Separator.log_onnxruntime_packagess#'"?"?@Q"R&*&C&CDY&Z#"&"?"? "N"&"?"?@V"W " . KK  PQhQpQpPqr s & 2 KK  TUpUxUxTyz { " . KK  PQhQpQpPqr s " . KK  UVmVuVuUvw x /rcfSn[R"5nURS5n[R"S5Ul[R R5(aURU5 SnO[[RS5(aQ[RRR5(a$URS:XaURU5 SnODUR(a3U(a,SSKnUR5(aUR!U5 SnU(d|UR"R%S 5 UR UlS /UlUR(d6U(dURS 5bUR"R%S 5 gggg) rFtorch_directmlcpuTmpsarmrNzANo hardware acceleration could be configured, running in CPU modeCPUExecutionProviderrzDirectML packages detected but DirectML is not enabled. Pass use_directml=True (or --use_directml on the CLI) to enable experimental DirectML acceleration.)rget_available_providersrprdevicercuda is_availableconfigure_cudarbackendsrr configure_mpsrr configure_dmlrcrrrr)rrhardware_acceleration_enabled ort_providershas_torch_dml_installedrs r8rSeparator.setup_torch_devicesY).%335 "&"?"?@P"Q % U 3 :: " " $ $    .,0 ) U^^U + +0B0B0O0O0Q0QVaVkVkotVt   } -,0 )   #: !**,,""=104-, KK  ` a $ 5 5D ,B+CD ($$*ATEbEbcyEzFG   zFG$-rcURRS5 [R"S5UlSU;a$URRS5 S/UlgURR S5 g)zT This method configures the CUDA device for PyTorch and ONNX Runtime, if available. z8CUDA is available in Torch, setting Torch device to CUDArCUDAExecutionProviderzFONNXruntime has CUDAExecutionProvider available, enabling accelerationzWCUDAExecutionProvider not available in ONNXruntime, so acceleration will NOT be enabledN)rcrrrrrrrrrs r8rSeparator.configure_cudasa ST!LL0 "m 3 KK  e f,C+DD ( KK   y zrcURRS5 [R"S5UlURUlSU;a$URRS5 S/UlgURRS5 g)zh This method configures the Apple Silicon MPS/CoreML device for PyTorch and ONNX Runtime, if available. z`Apple Silicon MPS/CoreML is available in Torch and processor is ARM, setting Torch device to MPSrCoreMLExecutionProviderzHONNXruntime has CoreMLExecutionProvider available, enabling accelerationzYCoreMLExecutionProvider not available in ONNXruntime, so acceleration will NOT be enabledN)rcrrrrrrrrr!s r8rSeparator.configure_mpsso {| % U 3 11 $ 5 KK  g h,E+FD ( KK   { |rcSSKnURRS5 UR5UlURUlSU;a$URRS5 S/UlgURRS5 g)zX This method configures the DirectML device for PyTorch and ONNX Runtime, if available. rNz@DirectML is available in Torch, setting Torch device to DirectMLDmlExecutionProviderzEONNXruntime has DmlExecutionProvider available, enabling accelerationzVDmlExecutionProvider not available in ONNXruntime, so acceleration will NOT be enabled)rrcrrrtorch_device_dmlrrr)rrrs r8rSeparator.configure_dmlsr  [\ . 5 5 7 11 !] 2 KK  d e,B+CD ( KK   x yrc[R"U5$![Ra" URR SUS35 gf=f)zh This method returns the package distribution for a given package name if installed, or None otherwise. zPython package: z not installedN)r distributionPackageNotFoundErrorrcr)r package_names r8rp"Separator.get_package_distributionsL ((6 6,,  KK   0nM N s3A Ac vURRSU35 Sn[RR U5n[ US5nX2:aUURRSUSUS35 [ R"UR55R5nO{X2- nURRSUSUSUS 35 URU[R5 [ R"UR55R5nS S S 5 URRS US W35 U$!,(df  N1=f![a URRS U35 e[ a'nURRSUSU35 eS nAff=f)z9 This method returns the MD5 hash of a given model file. zCalculating hash of model file i@rbz File size z < z, hashing entire file.z >= z , seeking to z and hashing remaining bytes.NzHash of model file z is zModel file not found at zError calculating hash for : )rcrr,r1getsizeopenhashlibmd5read hexdigestseekioSEEK_SETrrryrr)r model_path BYTES_TO_HASH file_sizer hash_valueseek_poses r8get_model_hashSeparator.get_model_hashs ;J<HI$   3Ij$'1,KK%% 9+SOe&fg!(QVVX!6!@!@!BJ )8HKK%% 9+T-P]^f]ghE'FGFF8R[[1!(QVVX!6!@!@!BJ( KK  2:,d:,O P (' !  KK   8 E F   KK   ;J>+ & & KK   7 }DWX Y  23%tK=HZ[\<<D#>   3 &"%h&6&6&:&:;KQ&O"P &9QUVLk4(A%22d2CE ''E 3GGEND)    !>seCTU]UiUiTjkl l )(s AD?? E cX [RRURS5nUR SU5 [ R "[USS95nURRS5 0n[R"SS5n[ R "U5nSSS5 URRS 5 US R5VVs0sHupgUR!S 5(dMXg_M! nnn0n UR5Hup[#SU R%55S5n U (dM/UR'U 05n U U R'S05U R'S/5U R'S5[)U R+55S.X'M [R"SS5n[ R "U5nSSS5 URRS5 0USEWSER5V Vs0sHnupU UUR'U05R'S05UR'U05R'S/5UR'U05R'S5U/S._Mp snn 0USEUSEUSER5V Vs0sHnupU UUR'U05R'S05UR'U05R'S/5UR'U05R'S5U/S._Mp snn U 0USEUSEUSEUSEUSER5V V s0sGHupU [#[-U R%555UR'[#[-U R%55505R'S05UR'[#[-U R%55505R'S/5UR'[#[-U R%55505R'S5[)U R%55[)U R+55-S._GM" sn n S.nU$!,(df  GN=f![ RaMnURRS [U535 URRS 5 SnAGN0SnAff=fs snnf!,(df  GN1=fs snn fs snn fs sn n f)a This method lists the supported model files for audio-separator, by fetching the same file UVR uses to list these. Also includes model performance scores where available. Example response object: { "MDX": { "MDX-Net Model VIP: UVR-MDX-NET-Inst_full_292": { "filename": "UVR-MDX-NET-Inst_full_292.onnx", "scores": { "vocals": { "SDR": 10.6497, "SIR": 20.3786, "SAR": 10.692, "ISR": 14.848 }, "instrumental": { "SDR": 15.2149, "SIR": 25.6075, "SAR": 17.1363, "ISR": 17.7893 } }, "download_files": [ "UVR-MDX-NET-Inst_full_292.onnx" ] } }, "Demucs": { "Demucs v4: htdemucs_ft": { "filename": "htdemucs_ft.yaml", "scores": { "vocals": { "SDR": 11.2685, "SIR": 21.257, "SAR": 11.0359, "ISR": 19.3753 }, "drums": { "SDR": 13.235, "SIR": 23.3053, "SAR": 13.0313, "ISR": 17.2889 }, "bass": { "SDR": 9.72743, "SIR": 19.5435, "SAR": 9.20801, "ISR": 13.5037 } }, "download_files": [ "https://dl.fbaipublicfiles.com/demucs/hybrid_transformer/f7e0c4bc-ba3fe64a.th", "https://dl.fbaipublicfiles.com/demucs/hybrid_transformer/d12395a8-e57c48e6.th", "https://dl.fbaipublicfiles.com/demucs/hybrid_transformer/92cfc3b6-ef3bcb9c.th", "https://dl.fbaipublicfiles.com/demucs/hybrid_transformer/04573f0d-f3cf25b2.th", "https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models/htdemucs_ft.yaml" ] } }, "MDXC": { "MDX23C Model: MDX23C-InstVoc HQ": { "filename": "MDX23C-8KFFT-InstVoc_HQ.ckpt", "scores": { "vocals": { "SDR": 11.9504, "SIR": 23.1166, "SAR": 12.093, "ISR": 15.4782 }, "instrumental": { "SDR": 16.3035, "SIR": 26.6161, "SAR": 18.5167, "ISR": 18.3939 } }, "download_files": [ "MDX23C-8KFFT-InstVoc_HQ.ckpt", "model_2_stem_full_band_8k.yaml" ] } } } zdownload_checks.jsonz]https://raw.githubusercontent.com/TRvlvr/application_data/main/filelists/download_checks.jsonutf-8encodingzUVR model download list loadedrzmodels-scores.jsonNzModel scores loadedzFailed to load model scores: zContinuing without model scoresdemucs_download_listz Demucs v4c3T# UHoRS5(dMUv M g7f).yamlN)r/).0r7s r8 7Separator.list_supported_model_files..sb|8GXGXY`Gahh|s( ( median_scoresstems target_stem)r7scoresrirjdownload_filesz models.jsonz!Audio-Separator model list loadedvr_download_listmdx_download_listmdx_download_vip_listmdx23c_download_listmdx23c_download_vip_listroformer_download_list)r\r[r]r^)r,r1r2rwr\rrr3rcrrrJSONDecodeErrorrstritems startswithnextrrvrvaluesiter)rdownload_checks_pathmodel_downloads_list model_scoresrr@keyvaluefiltered_demucs_v4 demucs_modelsrr6 yaml_filemodel_score_dataaudio_separator_models_listr7model_files_grouped_by_types r8list_supported_model_files$Separator.list_supported_model_filesCsIn "ww||D,?,?AWX ((*IK_ `#yy.BW)UV :<  C$$%68LMQR#yy| N KK   3 5 #3#7#7 #F&*5<<>&: ' # 6 !2M Ba*.))A, 'C =?'F)=>P)Q'FUprDVE'F'L'L'N 'NND (*..x<@@RTU)--h;??L#/#3#3Hb#A#E#Em#T'/j 'N $'y)=>Q)R'yVjlCWD'yHcdwHx'y'''A 'AND (*..x<@@RTU)--h;??L#/#3#3Hb#A#E#Em#T'/j 'A $$*+AB$*+EF$++CD$22HI $ 22JK $ %' $$KD $T%**,%7 8*..tD4F/GLPPQ`bde)--d4 3E.FKOOPWY[\#/#3#3Dejjl9K4Lb#Q#U#UVc#d&*5::<&84 ;O&O $-&' #P+*ONM## C KK  "?Ax H I KK   A B B C P$C B  s^9RR '#R TTTA5T8A5T D'T& RRS?2AS::S? TcUR(aEURRSURS35 URRS5 gg)z} This method prints a message to the user if they have downloaded a VIP model, reminding them to support Anjok07 on Patreon. z The model: 'zL' is a VIP model, intended by Anjok07 for access by paying subscribers only.zIf you are not already subscribed, please consider supporting the developer of UVR, Anjok07 by subscribing here: https://patreon.com/uvrN)rrcrrrs r8print_uvr_vip_messageSeparator.print_uvr_vip_messagesX  KK  ,t/G/G.HIU!V W KK  !k l !rc[RRURU5nUR 5nSnSnSnSnUR R SUS35 UR5GH~upU R5GHcupSU ;UlUR(aUOUn U SU:Xd XS ;dM8UR R S U 35 Xl UR5 U S Hn U RS 5(aQU RS 5S n[RRURU5nURX5 Mj[RRURU 5nUS:Xa4U RS5(aU nU SU 3nURUU5 MU S U 3nURUU5 M XXU4s s $ GM [!SUS35e![a7 UR R S5 US U 3nURUU5 Nf=f![a9 UR R S5 US U 3nURUU5 GMf=f)z This method downloads the model files for a given model filename, if they are not already present. Returns tuple of (model_filename, model_type, model_friendly_name, model_path, yaml_config_filename) zLhttps://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_modelsz:https://github.com/Anjok0109/ai_magic/releases/download/v5zVhttps://github.com/nomadkaraoke/python-audio-separator/releases/download/model-configsNzSearching for model_filename z! in supported_model_files_groupedVIPr7rlzFound matching model: http/r^rdz/mdx_model_data/mdx_c_configs/zHYAML config not found in UVR repo, trying audio-separator models repo...zBModel not found in UVR repo, trying audio-separator models repo...z Model file z# not found in supported model files)r,r1r2rwrrcrrurrrrvsplitr\r/rUr_)rrr;supported_model_files_groupedpublic_model_repo_url_prefixvip_model_repo_url_prefix&audio_separator_models_repo_url_prefixyaml_config_filename model_typerVr model_infomodel_repo_url_prefixfile_to_downloadr7 download_pathyaml_url download_urls r8download_model_filesSeparator.download_model_filess WW\\$"5"5.9IK (,(G(G(I%'u$$`!2J.# 9.9IIjkl#@"E"E"G J39<<>/#(-1D(D%EIEZEZ(A`|%j)^;~\lQm?mKK%%(>?R>S&TU/B,..0-77G,H(+66v>>'7'='=c'B2'FH,.GGLL9L9Lh,WM <<=M]$(* T5H5HJZ([ &/4D4M4Mg4V4V3C0Z.C-DDbcsbt+u $ @ @= Y %Z.C-DAFVEW+XL <<\=Y3-I>*7JXlllU4B#H\;~&66YZ[[#$0Z $ 1 12| }.T-UUVWgVh+i $ @ @= YZ ,Z KK--.rs.T-UUVWgVh+iL <<\=YYZs$-H I >I I >J J c$[RRU5(d+[RRURU5nOUnUR R SU35 [R"[USS9[RS9nUR R SU35 SSK J n U"5RU5n[RRU5R5nUcSU;aS US 'U$) z This method loads model-specific parameters from the YAML file for that model. The parameters in the YAML are critical to inferencing, as they need to match whatever was used during training. z%Loading model data from YAML at path r_r`)Loaderz"Model data loaded from YAML file: r?)ConfigurationNormalizerroformerT is_roformer)r,r1rxr2rwrcryamlrr3 FullLoader!roformer.configuration_normalizerrdetect_model_typebasenamer.)rrmodel_data_yaml_filepath model_datarconfigured_roformer_type yaml_filenames r8load_model_data_from_yaml#Separator.load_model_data_from_yaml6s ww~~233')ww||D4G4GI]'^ $'; $ ABZA[\]YYt$zlKLN#:#<#N#Nz#Z (()ABHHJ # /:3N(,J} %rcnSnUS3nUS3nURRS5 URU5nURRSUSU35 [RR UR S5nURRSU35 URX65 [RR UR S 5nURRS U35 URXG5 URRS 5 [R"[US S 95n[R"[US S 95n URRS5 [R"SS5n [R"U 5n SSS5 0UEW RS05En0U EU RS05En XY;aXn OXX;aXn O[SUS35eURRSUSU 35 U $!,(df  N=f)aB This method loads model-specific parameters from UVR model data files. These parameters are critical to inferencing using a given model, as they need to match whatever was used during training. The correct parameters are identified by calculating the hash of the model file and looking up the hash in the UVR data files. z>https://raw.githubusercontent.com/TRvlvr/application_data/mainz"/vr_model_data/model_data_new.jsonz#/mdx_model_data/model_data_new.jsonzQCalculating MD5 hash for model file to identify model parameters from UVR data...Model z has hash zvr_model_data.jsonzVR model data path set to zmdx_model_data.jsonzMDX model data path set to z@Loading MDX and VR model parameters from UVR model data files...r_r`zKLoading additional model parameters from audio-separator model data file...rzmodel-data.jsonN vr_model_datamdx_model_dataz0Unsupported Model File: parameters for MD5 hash z> could not be found in UVR model data file for MDX or VR arch.zModel data loaded using hash r1)rcrrAr,r1r2rwr\rrr3rrrvr_) rr;model_data_url_prefixvr_model_data_urlmdx_model_data_url model_hashvr_model_data_pathmdx_model_data_pathvr_model_data_objectmdx_model_data_objectraudio_separator_model_datars r8load_model_data_using_hash$Separator.load_model_data_using_hashOs2!a455WX 566YZ mn((4  F:,j EF WW\\$*=*=?ST 67I6JKL (():O ggll4+>+>@UV 78K7LMN (();Q \]#yy.@7)ST $ $/BW*U V gh  !24E F!)-1 &G o"6n:T:X:XYhjl:mn q#8 q_SUR@_SURB_SURD_SURF_URHURJURLURNS.EnS S!S"S#S$.nXpRP;dX;alS%UR'5;dU (aCU RSS&S'5(a,S(US)3nUR RUU5 [WU5e[WS*U35eUS+:Xa[XRZS,:a []S-5eUR R%S.US/X35 XR#S 5unn[^R`"S0U35n[cUU5nUR R%S1US/U35 U"XRPUS29n[cUS5S5n[gU5(a([cUS6S'5(dU"UR'55 [iUS75(aFURj(a5URm5nU(aUR RS8U35 UUlX@lU/UlXl Xl Xl Xl Xl UR R%S95 UR RS:[Rp"S;[Rr"[u[R"5U- 55535 g![\aenS%UR'5;dU (aDU RSS&S'5(a-S3US43nUR RUU5 [eU5UeeSnAff=f![na XPl X`l ef=f)"Q&&*>&:#'+N';$   #MdNbNbMc!de+A.N 3 3 ?DD_D_cqDq"0 $2#3D '+'G'GD $$($A$AD ! KK  vn%55gh i  >.)9=> $ 1 1 3"0'+'?'?$$($9$9!Z `d`y`y{IaJ ]N(;I]#44 '--c215J KK   ABUAVVdeodpq r!**733'1$#/!;;a!5!5!77J:>>ZginKoKo#PQ[P\]U"VIKK%%i0$Y//$'G |%TUUX%#*:*:W*D UVV KK   @ BO`OlNmn o&7&C&I&I#&N #K,,/WXcWd-efF%fj9O KK   Mj\Y[\k[lm n !0}ZsZst~Z"A%^5OQUVN''Hdfk0l0lz//12~':;;@`@`!/!J!J!L!KK$$'??O%PQ-5 78&4##6 0+>((8% 45 0z4;;WZ[_[l[l[nrG\GXHLI2J1KL MK !5!5!77J:>>ZginKoKo#I!Mm"nIKK%%i0&y1q8 & (D $$= !   s3J=W)U7&BW)7 W&A W!!W&&W))Xc UR(aIUR(dC[UR[5(a[ UR5S:d [ S5e[UR[5(Ga[ UR5S:Ga[U[5(a5[RRU5(dURX5$/n/n[U[5(aU/OUnUHTn[RRU5(aUR[U55 MCURU5 MV /nUH$nURURXb55 M& U(a [%Xt5eU$[U[5(aS[RRU5(d/UR R'S U35 UR)X5$[U[5(aU/n/n/nUHn[RRU5(aU[U5HDn UR R'S U 35 UR)X5n URU 5 MF M|UR R'S U35 UR)Xb5n URU 5 M U(a [%Xt5eU$![a>nUR R#SUSU3SS9 URXh45 SnAGMSnAff=f![a>nUR R#S U SU3SS9 URX45 SnAGM1SnAff=f![a>nUR R#S USU3SS9 URXh45 SnAGMSnAff=f) a4 Separates the audio file(s) into different stems (e.g., vocals, instruments) using the loaded model. This method takes the path to an audio file or a directory containing audio files, processes them through the loaded separation model, and returns the paths to the output files containing the separated audio stems. It handles the entire flow from loading the audio, running the separation, clearing up resources, and logging the process. Parameters: - audio_file_path (str or list): The path to the audio file or directory, or a list of paths. - custom_output_names (dict, optional): Custom names for the output files. Defaults to None. Returns: - output_files (list of str): Fully written output paths produced by inputs that completed their full stem set. Valid silent and near-silent stems are included. Raises: - InvalidAudioDataError: Decoded or generated audio data was empty, non-finite, or structurally invalid. - AudioExportError: An audio backend or filesystem failed to publish an output file. - BatchSeparationError: A list or directory input had one or more failures. All discoverable inputs are attempted first; inspect ``successful_files`` and the ordered ``failures`` list. A single file fails immediately with its original exception. Import the public exception types from ``audio_separator.separator``. rz]Initialization failed or model not loaded. Please load a model before attempting to separate.r?z Failed to process ensemble file r1Texc_infoNzProcessing file: zFailed to process file )rrrrrrr_rtr,r1isdir_separate_ensembleextendr9appendrrcrr rr_separate_file) raudio_file_pathcustom_output_namesensemble_inputsfailures input_pathsr1 output_filesr@ full_path files_outputs r8separateSeparator.separatese4!!t':':z$J]J]_c?d?dilmqnAnAjBEFjF|} } d))4 0 0S9L9L5MPQ5Q/3// o8V8V..TT OH/9/3/O/O?+UdK#77==&&#**+Ft+LM#**40 $ L'/ ''(?(?(Z[(*<BB  os + +BGGMM/4R4R KK  00AB C&&L L os + +./O $Dww}}T""! >[!/KK%%(HbQRPS&T_c%dOOTI../<%8 )),CI;bQRPS*T_c)d 778!/KK%%(?vRs&KVZ%[OOTI../sH* K=<"M"N= M2MM N2N  N O2OOc  URbSSKnUR"US9nSSKJn U"URUR 5nUR U5(a>UR RSUSSURS35 URX5$UR RS U35 [R"5nUR RS URS 35 UR RS URS 35 SnURn [UR SUR"5n [U S[%U 55n U [&:Xa7UR RS5 UR R)X5nOU [*:Xa]U S:waWUR RSU 5 [,R."U 5 UR R)X5nSSS5 O6UR RS5 UR R)X5n[0R2"5Sn /n SHn[UR U5"5 M! U (aeU b4[;U S5(a"U HupU R=SUSU35 M O/U Sup[;US5(aUR=SU35 UeUR?5 UR RS5 UR RS[R@"S[RB"[E[R"5U- 55535 U$!,(df  GNN=f![4a>nU R7X45 UR R9SUSU3SS9 SnAGMxSnAff=f![0R2"5Sn /n SHin[UR U5"5 M!![4a=nU R7X45 UR R9SUSU3SS9 SnAMcSnAff=f U (afU b5[;U S5(a#U HupU R=SUSU35 M ffU Sup[;US5(aUR=SU35 Uef=f)a  Internal method to handle separation for a single audio file. This method performs the actual separation process for a single audio file. It logs the start and end of the process, handles autocast if enabled, and ensures GPU cache is cleared after processing. Parameters: - audio_file_path (str): The path to the audio file. - custom_output_names (dict, optional): Custom names for the output files. Defaults to None. Returns: - output_files (list of str): A list containing the paths to the separated audio stem files. Nr)r1 AudioChunkerzFile duration z.1fzs exceeds chunk size zs, using chunked processingz1Starting separation process for audio_file_path: zNormalization threshold set to zC, waveform will be lowered to this max amplitude to avoid clipping.zAmplification threshold set to z?, waveform will be scaled up to this max amplitude if below it.rtypezUsing native float16 inference. privateuseonezUsing autocast inference on %s.zUsing float32 inference.r?)clear_gpu_cacheclear_file_specific_pathsz Cleanup step z failed: Tradd_notezCleanup step failed: zSeparation process completed.zSeparation duration: r)#rlibrosa get_duration(audio_separator.separator.audio_chunkingrrc should_chunkrr_process_with_chunkingrrrr}r~rrrrrtrrr autocast_modeautocastrrrrrrrrrrr)rrrrdurationrchunkerseparate_start_timerrinference_deviceinference_device_type active_errorcleanup_errors cleanup_name cleanup_errors r8rSeparator._separate_fileos    * ++AH M"4#6#6 DG##H--   >(3?TUYUhUhTijE"FG22?XX L_L]^_"//1 ;D!,LN P mD//>@!Q+|Z88;I7L(11M,yYfXg2hi!,LN P mD//>@ m"))<*GHKK%% l^9]O&\gk%lm!Q+|Z88;I7L(11M,yYfXg2hi.s-#i~SeS^f[MDIIe I##+D#E DXJOPh <"4#6#6 DG!--oHK(* $"$ !*;!7    #4QqSE3{;K:LBzl![\*.*=*='&*oo#&*#"*&&040C0C0N0N-5=''2"S#'#6#6z#BL),&2;<1H- I#%77#3#3I#> " /8 D (- AI+0 |([^[`N!+,RSS1::M0N0N,K.0B VN[$@AA0N  ))#ivi|i|i~#+.\]d\e,fgg(()9)9/)JKANIL#$<$A$A$CD '? 'J$+KK''*DYK(PQ'98K+K&9)&DO)2 2i[&BO ggll4??>OqQUQcQcQiQiQkPlZ [ww~~h'' !!$EhZ"PQ hd ;(e+B'&9O**9R++6+Jww~~h'' !!$EhZ"PQ hd ;(s@CU/ E.T5;=U/=A'U/$ U)5U)FU/51U&& U//AWc jURRSUS35 URU5upp4nUR5R S5(aUnUbUR U5nOUR U5n[U5nURRSUSUSUSUS 3 5 g) z: Downloads the model file without loading it into memory. zDownloading model rrdNzModel downloaded, type: z, friendly name: rz, model_data: z items)rcrrrr.r/rrr)rrrrr;rrmodel_data_dict_sizes r8download_model_and_data!Separator.download_model_and_data?s -n-=SAB\`\u\uwE]FY$7EY     & &w / /#-  +778LMJ88DJ": 3J5Separator.get_simplified_model_list..s !A$v,r)r}cJ>USSTnUcSOSUbU4$[S54$)Nr?rJrz-inf)float)itemsdr sort_by_lowers r8sort_key5Separator.get_simplified_model_list..sort_keys9q'%.7C!$A!COS__QVW]Q^__rT)r}reverse) rrurvrdictroundrr.r)rrG model_filessimplified_listrrVrdatar7rkrirjstems_with_scores stem_sdr_dictstem stem_scores stem_displayrWkv filtered_listrYrXs @r8get_simplified_model_list#Separator.get_simplified_model_listSs557 "-"3"3"5 J$lln  +(+1r)/R"hh}5 %'! " "D"(**T2"6K151DdV1:$L!+t44+9M#K$6:)00L>C51JK69 jjl3*00>6: jjl3"))2 %/3M),59Xi-A)=-#6D 'F?#8#8#:@VWXX:-F?#8#8#:;<<!/ 4 4 6 2A2G2G2I g2I$!Q]^_`e^fMfA2I g`F=#6#6#8hPTUVV!hs HHc SSKnSSKn[U[5(aU/n/nURnUR nUGHnUR RSU35 URSS9n UR RSU 35 0n URn UGHn UR RSU 35 URU 5 Xl UR(aXRl URUS5n /nU Hdn[RR!U5n["R$"SU5nU(aUR'S 5OS nUR)U5 Mf [+U5n[-S U55n[/X5HunnUR15nS U;aS U;a SU;aSnOYUS:Xa/US:Xa)U(a"SnUR RSUS35 O$U[2;a [2UnOUR55nUU ;a/U U'[RR7U5(aUO[RR9X5nU UR)U5 M Xl GM [;UR UR<UR>5n[RRA[RR!U55SnU RC5GHYunnUR RS[+U5SU35 /nSnUHwn[DRF"USURHS9unnURJS :XaUcS n[LRN"UU/5nOUcURPSnUR)U5 My URSU5n US :XaU RPSS :a U SS 2SS24n U(a UU;aUUn!OURT(aUSUSURT3n!O/n"UHhn#[RRAU#5Sn$SH)n%U$RWU%5(dMU$[+U%5Sn$ O U"R)U$SS5 Mj SR9U"5n&USUSU&3n!U!SURXR153n'UR(aXRl-URURl URR]U'U R^5 [RR9URU'5n(UR)U(5 GMjUR RaS U'35 [RR9URU'5n(SSK1n)SS!K2J3n* [iU R^5n+U*RkU+URlURnS"9n+[qU(S#5n,UR RS$U(S%35 U)RsU,U+URH5 SSS5 UR)U(5 GM\ X`lXplSUl [RR{U 5(a1UR RS*U 35 UR}U S+S,9 GMGM U$!Xl f=f!,(df  N=f![tan-UR RwS&URXS'U-S(35 U(RySS 5SS)-n([qU(S#5n,U)RsU,U+URH5 SSS5 Sn-A-GM(!,(df  Sn-A-GN;=fSn-A-ff=f!X`lXplSUl [RR{U 5(a/UR RS*U 35 UR}U S+S,9 ff=f)-z: Internal method to handle ensembling of multiple models. rNzEnsemble processing for file: zaudio-separator-ensemble-rz*Created temporary directory for ensemble: zProcessing with model: rr?rKc3|# UH2nSUR5;=(d UR5S;v M4 g7f)vocal)rN)r.)ress r8rf/Separator._separate_ensemble..s5-%5$qwwy0LAGGI4LL%5s:<rmleadbackingrrrMruDMapped 'other' → 'Instrumental' for 2-stem model (model produced: rz Ensembling z stems for type: F)monosrrz )_preset_)mel_band_roformer_melband_roformer_ bs_roformer_model_bs_roformer_z UVR-MDX-NET- UVR_MDXNET_ _z)_custom_ensemble_rzPNo model instance available to write ensembled audio. Using fallback writer for ) spec_utils)wavemax_peakmin_peak soundfilez'Attempting to write ensembled audio to rzError writing z format: z. Falling back to WAV.r)rTr)?rrrrtrrrcrrrrrtrrrr,r1rr#r$r%rranyzipr. STEM_NAME_MAPtitler'r2r rrr)rurrrndimnpasfortranarrayshapeensemblerrvr{r write_audioTrr$audio_separator.separator.uvr_lib_v5r{r normalizer}r~rrSr rrsplitrxr+).rrrrrroriginal_model_filenameoriginal_model_filenamesr1r, stems_by_typer3r model_stemsmodel_stem_namesr7r7r8r9num_model_stemshas_vocal_stem raw_stem_name lower_namer: ensemblerr? stem_paths waveformsoriginal_channelsspwavrz ensembled_wavrA model_slugsmfrr slugs_strrWfinal_output_pathsfr{ensemble_audiotemp_output_pathrequested_format_errors. r8rSeparator._separate_ensembles  os + +./O "&"5"5#'#7#7 #D KK  =dVD E ''/J'KH KK   J8*U V] @!# &*oo#&>NKK$$'>~>N%OPOON3'/O**9A++6*> '+&9&9$&E ,.()4I')ww'7'7 'BH$&IIox$HE:? AYI,33I> *5+..>*?),-%5-* 9J&*4z9QV_gqVq,4 !+w!6?a;OTb,: $ 1 14xzJyKKL3M!N!+}!<,9*,E ,9,?,?,A ( =;= i 846GGMM)4L4LyRTRYRYR^R^_gRsH))4;;HE%9[(+>k'?p&dkk43J3JDLaLab GG,,RWW-=-=d-CDQG -:-@-@-B)IzKK$${3z?2CCTU^T_%`a "I(,%(!(buAQAQ!RQ88q=0845 1"$"3"3S#J"?C.603 ! -!((-)%.$6$6y$AM)A--2E2Ea2H12L(5bqb!e(< +yB++;9=++6++77 ]__U,.GGLL+,V)$++,=> ++.~@KL-MN,.GGLL+,V).S)>})O)3)=)=!/%)%A%A%)%A%A*>* ]!34E{!SWg $ 1 14[\m[nnq2r s ")9>4K[K[ \"T%++,=>_.Cf'>#'?$'+#77>>(++KK%%(I(&TUMM($M?,E$LI+>T"T!S 0] KK--"01C1C0DINdMee{ |1B0H0Ha0PQR0SV\0\-!34E{!SWg ")9>4K[K[ \"T!S!S!S!S ]'>#'?$'+#77>>(++KK%%(I(&TUMM($M?,s A*^8F[ G'^5F^ [8=['[8^[$$^' [5 1[84^5[88 ^A^]? 0^8^? ^ ^ ^^^^A(`)%rrrrr~rrrrrrrfrhrercrwrrrrrr}rr|rtr{rrrrr(rrrrrr)rFr+))rb __module__ __qualname____firstlineno____doc__r`INFOrpropertyrrrrrrrrrrrrrrrprAr\rrrrrrrrrrErrtrir__static_attributes__rrr8r;r;@s7v,,5 # #"&\]qvw!"3aW\uzWZpuv'0A$dhi%(^bosEFG5_8BII TT ! %N /-f"$ y$!F { } z!Fm0h+Tl?\B21fNM`[zM^ >@urr;).rrrrr,rrrrr`rnr9r#rnumpyrtypingrr4rrrNrtorch.amp.autocast_modeamprrrr"audio_separator.separator.audio_iorr#audio_separator.separator.ensemblerr $audio_separator.separator.exceptionsr r r *audio_separator.separator.execution_policyr rrrr0r9r;rrr8rsVa)     //X9nnRR hN N~   W    W F h W=y W =!"&#$#&' ,a3HHr