Vh &dZddlZddlmZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZmZddlmZejeZd Zeejd z Zgd Zd Zd ZdZdeje e ffdZ! dde dej"efdZ#dZ$dS)zLoading pretrained models. N)Path)fatalbold)HDemucs) RemoteRepo LocalRepo ModelOnlyRepo BagOnlyRepo AnyModelRepoModelLoadingError) _check_diffqz&https://dl.fbaipublicfiles.com/demucs/remote)drumsbassothervocalshtdemucsc2tdt}|S)N)channelssources)rSOURCES)models W/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/demucs/pretrained.pydemucs_unittestrs Q 0 0 0E Lc|d}|ddd|dddd |d td dS)NF)requiredz-sz--sigzLocally trained XP signature.)helpz-nz--namez8Pretrained model name or signature. Default is htdemucs.)defaultr z--repoz9Folder containing all pre-trained models for use with -n.)typer )add_mutually_exclusive_group add_argumentr)parsergroups radd_model_flagsr' s  / / / ? ?E tW+JKKK tXtVXXX tXZZZZZrreturncd}i}|dD]}|}|dr,|dr/|ddd}p|ddd}||vsJt|z|z||<|S) N #zroot::r-r) read_textsplitstrip startswithROOT_URL)remote_file_listrootmodelslinesigs r_parse_remote_filesr9)sD "F **,,22488 1 1zz|| ??3   1  __W % % 1::c1%%a(..00DD**S!$$Q'Cf$$$$"T/D0F3KK Mrnamerepoc |dkrtS|??'' {J77{{}} < T::: ; ; ;t__ tZ00J11H""4((  chqk ! ! NNN   JJLLL Ls)B?? C, C''C,c|j)t|_ttddt |j|jS)z8 Load local model package or pre-trained model. Nz?Important: the default model was recently changed to `htdemucs`zthe latest Hybrid Transformer Demucs model. In some cases, this model can actually perform worse than previous models. To get back the old default model use `-n mdx_extra_q`.)r:r;)r: DEFAULT_MODELprintrr@r;)rBs rget_model_from_argsrKVsN y!  dTUU& ' ' ' $)$) 4 4 44r)N)%__doc__loggingpathlibrtypingtpdora.logrrhdemucsrr;rr r r r r statesr getLogger__name__loggerr3__file__parentr>rrIrr'Dictstrr9Optionalr@rKrrr]st  dddddddddddddddd  8 $ $ 3d8nn#h. . . .  ZZZ RWS#X->    ")-CK%: 5 5 5 5 5r