ó
    4=ƒjÛ,  ã                   ó¼  • S SK r S SKrS SKrS SKrS SKrS SKJrJr  S SK	J
r
JrJrJrJr  S SKJr  S SKJrJrJrJr  \R,                  R/                  \R,                  R1                  \R,                  R3                  \5      5      S5      r\R,                  R/                  \R,                  R1                  \R,                  R3                  \5      5      S5      rSr\R,                  R/                  \S	5      r\R,                  R/                  \S
5      rS\ S\RB                  4S jr"S\#S\#S\#S\4S jr$S\#S\%S\&\%   4S jr'S\#S\ S\ 4S jr(S r)\*S:X  a  \)" 5         gg)é    N)ÚJSON_EXTÚVAE_EXT)ÚDEVICEÚdecode_codesÚn_render_framesÚ	predict_hÚpublic_warmup)ÚHiddenEncoder)Ú	CppOracleÚLocalSearcherÚPublishedStackÚchunk_startsz..Údatasetsz/tmp/minimax-refine-resultszbuild/mm-synthÚmodelsÚrequestÚreturnc                 ó  • [         R                  " U S   R                  S5      [         R                  S9nUR                  S-  (       a  [        S5      e[        R                  " UR                  SS5      5      R                  [        5      $ )NÚaudio_codesÚ,)Údtypeé   z(audio_codes must contain 8 IDs per frameéÿÿÿÿ)ÚnpÚarrayÚsplitÚint64ÚsizeÚ
ValueErrorÚtorchÚ
from_numpyÚreshapeÚtor   )r   Úvalss     Útraining/refine-e2e.pyÚrequest_codesr%       sd   € Ü�8Š8�G˜MÑ*×0Ñ0°Ó5¼R¿X¹XÑF€DØ‡y�y�1‡}ÜÐCÓDÐDÜ×Ò˜DŸL™L¨¨QÓ/Ó0×3Ñ3´FÓ;Ð;ó    ÚcorpusÚbaseÚckptÚstackc           	      ó<  • [         R                  R                  X[        -   5      n[         R                  R                  X[        -   5      n[        U5       n[        R                  " U5      nS S S 5        WR                  SS 5        SUS'   [        U[        [        SU 35      n[        [        UR                  SS5      5      S-  5      n	[        XY5      u  p«[!        5       R#                  [$        5      R'                  5       n[(        R                  " U[$        S9nUR+                  US	   5        [-        S
U SUS    SUS   S 3SS9  [/        XÅXº5      n[1        UR2                  UR4                  U5      u  nn[(        R6                  " U
S-   S4[$        [(        R8                  S9nUUS'   UUSS 2S4'   UUSS 2SS 24'   SR                  S UR;                  5       R=                  5       R?                  5        5       5      US'   UU4$ ! , (       d  f       GN¤= f)Nr   Úwav16Úoutput_formatz/tmp/minimax-warmup-Údurationg      N@g      9@)Úmap_locationÚmodelz[Init] z epoch Úepochz val Úval_lossz.4fT©Úflushé   r   )Údevicer   r   r   c              3   óJ   #   • U  H  n[        [        U5      5      v •  M     g 7f©N)ÚstrÚint©Ú.0Úvs     r$   Ú	<genexpr>Ú"initial_request.<locals>.<genexpr>@   s   é € Ð%YÒ;X°a¤c¬#¨a«&§k kÒ;Xùs   ‚!#) ÚosÚpathÚjoinr   r   ÚopenÚjsonÚloadÚpopr	   ÚMM_SYNTHÚ
MODELS_DIRr:   ÚfloatÚgetr   r
   r"   r   Úevalr   Úload_state_dictÚprintr   r   ÚlmÚdepthÚemptyÚlongÚcpuÚnumpyÚflatten)r'   r(   r)   r*   Úsource_pathÚvae_pathÚfr   ÚwarmupÚ
max_framesÚn_framesÚstartsr0   ÚstateÚhÚsemÚacÚcodess                     r$   Úinitial_requestra   '   sÆ  € Ü—'‘'—,‘,˜v¬h¡Ó7€KÜ�w‰w�|‰|˜F¬7¡NÓ3€HÜ	ˆkÔ	˜aÜ—)’)˜A“,ˆ÷ 
ð
 ‡K�K�˜tÔ$Ø&€GˆOÑÜØ”œ:Ð)=¸d¸VÐ'DóF€Fä”U˜7Ÿ;™; z°4Ó8Ó9¸DÑ@ÓA€JÜ& xÓ<Ñ€Hä‹O×ÑœvÓ&×+Ñ+Ó-€EÜ�JŠJ�t¬&Ñ1€EØ	×Ñ˜% ™.Ô)Ü	ˆG�D�6˜  w¡Ð 0°°e¸JÑ6GÈÐ5LÐ
MÐUYÒZÜ�% 6Ó4€AÜ˜5Ÿ8™8 U§[¡[°!Ó4�G€CˆÜ�KŠK˜ A™ qÐ)´&ÄÇ
Á
ÑK€EØ€Eˆ!�HØ€Eˆ!‰"ˆaˆ%�LØ€Eˆ!‰"ˆa‰bˆ&�MØ ŸX™XÑ%Y¸5¿9¹9»;×;LÑ;LÓ;N×;VÑ;VÔ;XÓ%YÓY€GˆMÑØ�Eˆ>Ð÷/ 
Ö	ús   ÁHÈ
HÚvalueÚ	n_windowsc                 ó  • U S:X  a  [        [        U5      5      $ U R                  S5       Vs/ s H$  o"R                  5       (       d  M  [	        U5      PM&     nnU H"  nUS:  d  XA:¼  d  M  [        SU SUS-
   35      e   U$ s  snf )NÚallr   r   zwindow z outside 0..r5   )ÚlistÚranger   Ústripr:   r   )rb   rc   r=   ÚoutÚks        r$   Úparse_windowsrk   D   s�   € Ø�ƒ~Ü”E˜)Ó$Ó%Ð%Ø Ÿ;™; sÔ+Ó
9Ò+�a¯w©w¯y‹6Œ3ˆqŽ6Ñ+€CÐ
9ÛˆØˆq‹5�A•NÜ˜w q c¨°iÀ!±m°_ÐEÓFÐFñ ð €Jùò	 :s   ®BÁ
BrA   Úmetac                 ó  • [         R                  " [         R                  R                  [         R                  R	                  U 5      5      SS9  [        U S5       n[        R                  " XSS9  UR                  S5        S S S 5        [         R                  R                  U 5      u  pE[        US-   S5       n[        R                  " X#SS9  UR                  S5        S S S 5        g ! , (       d  f       Nn= f! , (       d  f       g = f)NT)Úexist_okÚwé   )ÚindentÚ
ú
.meta.json)
r@   ÚmakedirsrA   ÚdirnameÚabspathrC   rD   ÚdumpÚwriteÚsplitext)rA   r   rl   rW   ÚstemÚ_s         r$   Úsave_resultr|   N   s®   € Ü‡K‚K”—‘—‘¤§¡§¡°Ó 5Ó6ÀÒFÜ	ˆd�CŒ˜AÜ�	Š	�' QÒ'Ø	�‰�Œ÷ 
ô �g‰g×Ñ˜tÓ$�G€DÜ	ˆd�\Ñ! 3Ô	'¨1Ü�	Š	�$ !Ò$Ø	�‰�Œ÷ 
(Ð	'÷	 
�ú÷ 
(Õ	'ús   Á'C+Â;'C<Ã+
C9Ã<
D
c                  ó’  • [         R                  " 5       n U R                  S5        U R                  S5        U R                  5       nUR                  SSS9  UR                  S[        R
                  R                  [        R
                  R                  [        5      S5      S9  U R                  S	5        U R                  S
5        U R                  S[        SS9  U R                  SSSS9  U R                  SSS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[        SS9  U R                  S[         R                  SS 9  U R                  5       n[        R
                  R                  [        UR                  5      n[        R
                  R                  X2R                  S!-   5      nUR                   =(       d>    [        R
                  R                  ["        UR                   S"UR                   S#35      nUR$                  =(       d    S$UR                   S"UR                   3n['        5       nUR(                  (       aá  [+        UR(                  5       n[,        R.                  " U5      n	S S S 5        S%W	S&'   [1        U	5      n
UR(                  nS n[        R
                  R3                  UR(                  5      S'   S(-   n[        R
                  R5                  U5      (       a;  [+        U5       n[,        R.                  " U5      nS S S 5        WR7                  S)5      nO1[9        X2R                  UR:                  U5      u  pšUR:                  nSn[        U	R7                  S*S+5      5      S':  a  [=        S,5      e[?        [@        XF5      nURC                  U	S-SS.9nURD                  n[G        [I        U
RJ                  S'   S-
  5      5      n[M        URN                  U5      nURP                  RS                  S/5       Vs/ s H*  nURU                  5       (       d  M  URU                  5       PM,     nn[W        S0 U 5       5      (       a  [=        S15      e/ n0 nUR                  UR                  UUS2S3UUU/ S4.
n[Y        XYU5        [[        S5US6 S7U S83SS99  Sn[]        UR^                  5       GH’  nS:n[[        S;US-    S<UR^                   3SS99  U GHj  nU GH_  n[[        S=U S>U 3SS99  [a        XyUURb                  URd                  S?9nUS@:X  a2  URg                  U
UURh                  URj                  URl                  SA9nO<URo                  U
UURp                  URh                  URj                  URl                  SB9nU(       d  [[        SCSS99  M»  SDUS-    SEU S"U 3n URs                  XšUUU URt                  SF9u  n!n"n#n$n%U$ V&s1 s H  n&U&Rv                  U&Rx                  4iM     n'n&U H(  n(U(Rv                  U(Rx                  4n)U)U';  d  M#  U(UU)'   M*     U%(       d  [[        SGSS99  GMV  Un*U!U"U#np©URC                  U	SHUSI 3SS.9nURD                  nUS-  nSnUS-   UUU*UU$ V&s/ s H  n&U&Rz                  PM     sn&SJ.n+UR}                  U+5        [        U5       HB  n)UU)   n([        U
U(Rv                  S-   U(Rx                  4   5      U(R€                  :w  d  M?  UU)	 MD     UUSK'   URƒ                  5        V&s/ s H  n&U&Rz                  PM     sn&USL'   [Y        XYU5        [[        SMU*S6 SNUS6 3SS99  GMb     GMm     UR„                  (       GaÙ  U(       GaÑ  [        U5       HB  n)UU)   n([        U
U(Rv                  S-   U(Rx                  4   5      U(R€                  :w  d  M?  UU)	 MD     [‡        URƒ                  5       SO SSP9n,U,S URl                   n,U,(       GaO  [[        SQ[G        U,5       SR3SS99  URs                  XšU,USDUS-    SS3URt                  SF9u  n!n"n#n$n%U%(       aÈ  Un*U!U"U#np©URC                  U	SHUSI 3SS.9nURD                  nUS-  nSnUR}                  US-   S SLU*UU$ V&s/ s H  n&U&Rz                  PM     sn&SJ.5        [        U5       HB  n)UU)   n([        U
U(Rv                  S-   U(Rx                  4   5      U(R€                  :w  d  M?  UU)	 MD     [[        STU*S6 SNUS6 3SS99  UUSK'   URƒ                  5        V&s/ s H  n&U&Rz                  PM     sn&USL'   [Y        XYU5        U(       a  GM‰  [[        SUSS99    O   UUSK'   URƒ                  5        V&s/ s H  n&U&Rz                  PM     sn&USL'   [Y        XYU5        [[        SVUS6 3SS99  [[        SWU 3SS99  g ! , (       d  f       GN<= f! , (       d  f       GN³= fs  snf s  sn&f s  sn&f s  sn&f s  sn&f s  sn&f s  sn&f )XNÚdatasetr(   z	--requestz7starting request; sibling metadata preserves provenance)Úhelpz--ckptzcheckpoints/v3/best.pt)Údefaultz--outputz
--work-dirz--sweepsr5   )Útyper€   z	--windowsre   z)all or comma-separated C++ window indices)r€   r   z--modeszacoustic,semanticz--coordsé   z--per-coordinatez--oracle-movesz--semantic-top-ké2   z--parity-ming+‡ÙÎ÷ï?z--min-global-gaing        z--revisit-archiveT)Úactionr€   z.wavÚ-z-refined.jsonz/tmp/minimax-refine-r,   r-   r   rs   Ú	leak_freeÚseedr   z1refinement requires a fixed non-negative DiT seedz
anchor-000)rw   r   c              3   ó*   #   • U  H	  oS ;  v •  M     g7f))ÚacousticÚsemanticN© r;   s     r$   r>   Úmain.<locals>.<genexpr>Ž   s   é € Ð
<²e°Ð,Ö,²eùs   ‚z&--modes accepts only acoustic,semanticz&published C++ AR from prompt + lm_seedz&final shifted 128-frame encoder window)
r~   r(   Úsourcer†   rX   ÚtailÚinitial_scoreÚscoreÚhistoryÚarchivez[Start] full-song STFT z.9fz, z windowsr3   Fz[Sweep] Ú/z[Search] window Ú )Ú
parity_minr‰   )Ún_coordsÚ	per_coordÚ	max_moves)Útop_kr–   r—   r˜   z![Search] no local improving cellsÚsz-w)Úmin_gainz[Oracle] rejected all subsetszanchor-Ú03d)ÚsweepÚwindowÚmodeÚbeforeÚafterÚmovesr�   r’   z	[Accept] z -> c                 ó   • U R                   $ r8   )Úlocal_score)Úms    r$   Ú<lambda>Úmain.<locals>.<lambda>â   s   € ¸Q¿]º]r&   )ÚkeyÚreversez[Archive] retesting z rejected movesz-archivez
[Archive] z'[Done] sweep made no global improvementz[Done] full-song STFT z[Done] request )DÚargparseÚArgumentParserÚadd_argumentÚadd_mutually_exclusive_groupr@   rA   rB   ru   Ú__file__r:   rI   ÚBooleanOptionalActionÚ
parse_argsÚDATASETS_DIRr~   r(   ÚoutputÚRESULTS_DIRÚwork_dirr   r   rC   rD   rE   r%   ry   ÚexistsrJ   ra   r)   r   r   ÚROOTÚrenderr�   Úlenr   Úshaperk   ÚwindowsÚmodesr   rh   Úanyr|   rM   rg   Úsweepsr   Údump_dirr•   Úacoustic_movesÚcoordsÚper_coordinateÚoracle_movesÚsemantic_movesÚsemantic_top_kÚsubset_searchÚmin_global_gainÚframeÚcodebookÚ__dict__Úappendrf   ÚoldÚvaluesÚrevisit_archiveÚsorted)-Úapr�   Úargsr'   Úref_pathr²   r´   r*   rW   r   r`   Úsource_descÚsource_leak_freeÚ	meta_pathÚsource_metaÚoracleÚanchorr�   rc   rº   r=   r»   r‘   r’   rl   Úserialr�   Úchangedrj   rŸ   Úsearcherr¢   ÚtagÚnew_requestÚ	new_codesÚ	new_scoreÚacceptedÚokr¥   Úaccepted_coordsÚmover¨   r    ÚentryÚrevisits-                                                r$   Úmainrå   Y   st	  € Ü	×	 Ò	 Ó	"€BØ‡O�O�IÔØ‡O�O�FÔØ×,Ñ,Ó.€FØ
×Ñ˜Ð*cÐÑdØ
×Ñ˜¬"¯'©'¯,©,´r·w±w·±ÄxÓ7PÐRjÓ*kÐÑlØ‡O�O�JÔØ‡O�O�LÔ!Ø‡O�O�J¤S°!€OÑ4Ø‡O�O�K¨Ð5`€OÑaØ‡O�O�IÐ':€OÑ;Ø‡O�O�J¤S°!€OÑ4Ø‡O�OÐ&¬S¸!€OÑ<Ø‡O�OÐ$¬3¸€OÑ:Ø‡O�OÐ&¬S¸"€OÑ=Ø‡O�O�N¬¸€OÑ>Ø‡O�OÐ'¬e¸S€OÑAØ‡O�OÐ'´×0NÑ0NÐX\€OÑ]Ø�=‰=‹?€Dä�W‰W�\‰\œ,¨¯©Ó5€FÜ�w‰w�|‰|˜F§I¡I°Ñ$6Ó7€HØ�[‰[×aœBŸG™GŸL™L¬¸¿¹¸ÀaÈÏ	É	À{ÐR_Ð6`Óa€FØ�}‰}×QÐ"6°t·|±|°nÀAÀdÇiÁiÀ[Ð Q€HäÓ€EØ‡|‡|Ü�$—,‘,Ô 1Ü—i’i “lˆG÷  à#*ˆ�Ñ Ü˜gÓ&ˆØ—l‘lˆØÐÜ—G‘G×$Ñ$ T§\¡\Ó2°1Ñ5¸ÑDˆ	Ü�7‰7�>‰>˜)×$Ñ$Ü�i” AÜ"Ÿiši¨›l�÷ !à*Ÿ™¨{Ó;Ðøä(¨·±¸D¿I¹IÀuÓM‰ˆØ—i‘iˆØÐä
ˆ7�;‰;�v˜rÓ"Ó# aÓ'ÜÐLÓMÐMä”t˜XÓ0€FØ�]‰]˜7 L°tˆ]Ð<€FØ�L‰L€EÜ”L §¡¨Q¡°!Ñ!3Ó4Ó5€IÜ˜DŸL™L¨)Ó4€GØ $§
¡
× 0Ñ 0°Ô 5ÓCÒ 5˜1¸¿¹¿‹YˆQ�W‰WŽYÑ 5€EÐCÜ
Ñ
<±eÓ
<×<Ñ<ÜÐAÓBÐBà€GØ€Gà—<‘<Ø—	‘	ØØ%Ø:Ø8ØØØØñ€Dô � Ô&Ü	Ð# E¨# ;¨b°°¸8Ð
DÈDÒQà€FÜ�t—{‘{×#ˆØˆÜ�˜ ™˜ 1 T§[¡[ MÐ2¸$Ò?ÜˆAÜ�ÜÐ(¨¨¨1¨T¨FÐ3¸4Ò@Ü(Ø H¨f¯o©oÈ$Ï/É/ñ[�à˜:Ó%Ø$×3Ñ3Ø˜q¨4¯;©;Ø"&×"5Ñ"5À×ARÑARð 4ð T‘Eð %×3Ñ3Ø˜q¨×(;Ñ(;Ø!%§¡¸×8KÑ8KØ"&×"3Ñ"3ð 4ð 5�Eö ÜÐ=ÀTÒJÙà˜% !™)˜ B q c¨¨4¨&Ð1�ØBH×BVÑBVØ E¨5°#À×@TÑ@Tð CWð CVÑ?�˜Y¨	°8¸RáBJÓ"KÂ(¸Q A§G¡G¨Q¯Z©ZÓ#8Á(�Ð"KÛ!�DØŸ:™: t§}¡}Ð5�CØ /Õ1Ø'+˜ ›ñ "ö ÜÐ9ÀÒFÚà�Ø(3°YÀ	 ˜ØŸ™ w°'¸&À¸Ð0FÈT˜ÐR�àŸ™�Ø˜!‘�Ø�à" Q™YØØ Ø$Ø"Ù2:Ó;²(¨Q˜aŸjœj±(Ñ;ñ�ð —‘˜uÔ%Ü ž=�CØ" 3™<�DÜ˜5 §¡¨a¡°·±Ð!>Ñ?Ó@ÀDÇHÁHÕLØ# CšLñ )ð !&��W‘Ø7>·~±~Ô7GÓ"HÒ7G°! 1§:¤:Ñ7GÑ"H��Y‘Ü˜F¨TÔ2Ü˜	 &¨ ¨T°%¸°Ð=ÀTÕJôi ñ ðn ××Ð§GÜ˜G–}�Ø˜s‘|�Ü�u˜TŸZ™Z¨!™^¨T¯]©]Ð:Ñ;Ó<ÀÇÁÕHØ šñ %ô ˜WŸ^™^Ó-Ñ3JÐTXÑYˆGØÐ0˜t×0Ñ0Ð1ˆGßÜÐ,¬S°«\¨N¸/ÐJÐRVÒWØBH×BVÑBVØ G¨U°a¸À¹	°{À(Ð4KØ!×1Ñ1ð CWð C3Ñ?�˜Y¨	°8¸Rö Ø"�FØ,7¸ÀI E˜UØ#Ÿ]™]¨7°g¸fÀS¸\Ð4JÐQU˜]ÐV�FØ"ŸL™L�EØ˜a‘K�FØ"�GØ—N‘NØ!&¨¡Ø"&Ø )Ø"(Ø!&Ù6>Ó!?²h° !§*¤*±hÑ!?ñ$ô ô  $ Gž}˜Ø& s™|˜Ü˜u T§Z¡Z°!¡^°T·]±]Ð%BÑCÓDÈÏÉÕPØ '¨¢ñ  -ô ˜J v¨c l°$°u¸S°kÐBÈ$ÒOØ %��W‘Ø7>·~±~Ô7GÓ"HÒ7G°! 1§:¤:Ñ7GÑ"H��Y‘Ü˜F¨TÔ2ç‰wÜÐ;À4ÒHÙñA $ðD €Dˆ�MØ+2¯>©>Ô+;Ó<Ò+; a�q—z”zÑ+;Ñ<€Dˆ�OÜ� Ô&Ü	Ð" 5¨ +Ð
.°dÒ;Ü	ˆO˜F˜8Ð
$¨DÓ1÷i  Öú÷ !–üò  Dùò\ #Lùò, <ùò #Iùò: "@ùò #Iùò =sH   Ê6hÍhÑ!h&Ñ>h&Ø+"h+
Ûh0Ý h5
ãh:å'h?æ>iè
hè
h#Ú__main__)+rª   rD   r@   rS   r   r   r~   r   r   Úe2er   r   r   r   r	   r0   r
   Úrefiner   r   r   r   rA   rB   ru   rv   r®   r¶   r±   r³   rG   rH   ÚdictÚTensorr%   r9   ra   r:   rf   rk   r|   rå   Ú__name__r‹   r&   r$   Ú<module>rì      sA  ðó Û Û 	ã Û ç %÷ õ  å ß IÓ Ià�w‰w�|‰|˜BŸG™GŸO™O¨B¯G©G¯O©O¸HÓ,EÓFÈÓM€Ø�w‰w�|‰|˜BŸG™GŸO™O¨B¯G©G¯O©O¸HÓ,EÓFÈ
ÓS€Ø,€Ø�w‰w�|‰|˜DÐ"2Ó3€Ø�w‰w�|‰|˜D (Ó+€
ð<˜4ð < E§L¡Lô <ð˜Cð  sð °#ð ¸nô ð:˜ð ¨ð °°c±ô ð�cð  Dð °ô òp2ðf ˆzÓÙ…Fð r&   