]i :SrSSKrSSKJr SSKrSSKJrJr SSK J r SSK J r J r JrJrJrJr SSKJr \R("\5rS r\"\5R2S - r/S QrS rS rSrS\R>\ \ 44Sjr!SS\ S\RD\4Sjjr#Sr$g)zLoading pretrained models. N)Path)fatalbold)HDemucs) RemoteRepo LocalRepo ModelOnlyRepo BagOnlyRepo AnyModelRepoModelLoadingError) _check_diffqz&https://dl.fbaipublicfiles.com/demucs/remote)drumsbassothervocalshtdemucsc"[S[S9nU$)N)channelssources)rSOURCES)models N/mnt/rpi/tmp/demucs-venv-sys/lib/python3.13/site-packages/demucs/pretrained.pydemucs_unittestrs Q 0E LcURSS9nURSSSS9 URSSSS S 9 URS [S S 9 g)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' sc  / / / ?E tW+JK tXtVX tXZrreturnctSn0nUR5RS5HnUR5nURS5(aM+URS5(a%URSS5SR5nMfURSS5SnXB;de[U-U-X$'M U$) N #zroot::r-r) read_textsplitstrip startswithROOT_URL)remote_file_listrootmodelslinesigs r_parse_remote_filesr9)sD "F **,2248zz| ??3    __W % %::c1%a(..0D**S!$Q'C$ $$"T/D0FK9 MrnamerepocUS:Xa [5$Uc.[[S- 5n[U5n[ [U5nO9UR 5(d[ US35 [U5n[ X5n[X45nURU5nUR5 U$![a#nSURS;a [5 eSnAff=f)z`name` must be a bag of models name or a pretrained signature from the remote AWS model repo or the specified local repo if `repo` is not None. rNz files.txtz must exist and be a directory.diffqr)rr9 REMOTE_ROOTrr is_dirrr r get_model ImportErrorargsreval)r:r;r6 model_repobag_repoany_reporexcs rr@r@9s       |$[;%>?' {J7{{}} TF9: ;t_ t0J1H""4(  JJL L  chhqk ! N sB** C4CCcURc [Ul[[S5S5 [ URUR S9$)z0 Load local model package or pre-trained model. z?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_argsrKVsA yy!  dTU& ' $))$)) 44r)N)%__doc__loggingpathlibrtypingtpdora.logrrhdemucsrr;rr r r r r statesr getLogger__name__loggerr3__file__parentr>rrIrr'Dictstrr9Optionalr@rKrrr]s  dd   8 $ 38n##h. .  Z RWWS#X-> ")-CKK%: 5r