
    ni                        S r SSKrSSKrS\R                  S'   SSKrSSKJr  SSKrSSKrSSK	J
r  SSKJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1  \Rd                  " S5         " S	 S
/\P\P\P\P\P\P\P\P\P\P\P\P\P\P\P\P\P\P\P\ P\!P\"P\#P\$P\%P\&P\'P\(P\)P\*P\+P\,P\-P\.P\/P\0P\1P76 r3g)zm
Business Logic Handler
Encapsulates all data processing and business logic as a bridge between model and UI
    NfalseTOKENIZERS_PARALLELISM)Optional)get_global_gpu_config)%AudioCodesMixinBatchPrepMixinConditioningBatchMixinConditioningEmbedMixinConditioningMaskMixinConditioningTargetMixinConditioningTextMixinDiffusionMixinGenerateMusicDecodeMixinGenerateMusicExecuteMixinGenerateMusicMixinGenerateMusicPayloadMixinGenerateMusicRequestMixinInitServiceMixinIoAudioMixinLyricScoreMixinLyricTimestampMixinLoraManagerMixinMemoryUtilsMixinMetadataMixinMlxDitInitMixinMlxVaeDecodeNativeMixinMlxVaeEncodeNativeMixinMlxVaeInitMixinPaddingMixinProgressMixinPromptMixinServiceGenerateMixinTrainingPresetMixinTaskUtilsMixinVaeDecodeChunksMixinVaeDecodeMixinVaeEncodeChunksMixinVaeEncodeMixinServiceGenerateRequestMixinServiceGenerateExecuteMixinServiceGenerateOutputsMixinignorec                       \ rS rSrSrS rSrg)AceStepHandler>   zACE-Step Business Logic Handlerc                 $   SU l         SU l        SU l        [        R                  U l        SU l        SU l        SU l        SU l	        SU l
        SU l        SU l        SSKJn  [        U5      U l        SU l        SU l        SU l        SU l        [*        R,                  R/                  S	S
5      R1                  5       S;   =(       d&    [3        [4        R6                  SS 5      " 5       (       + U l        [*        R,                  R/                  SS
5      R1                  5       S;   U l        SU l        [>        R@                  " 5       U l!        S/ 0U l"        [*        RF                  RI                  U RK                  5       SSS5      U l&        U RO                  5         SU l(        SU l)        SU l*        SU l+        SU l,        SU l-        0 U l.        0 U l/        SU l0        SU l1        SU l2        SU l3        SU l4        SU l5        [m        5       Rn                  U l7        g)zFInitialize runtime model handles, feature flags, and generation state.Ncpui     r   )_DEFAULT_LAYERS_CONFIGFg        ACESTEP_DISABLE_TQDM )1trueyesisattyc                      g)NF r;       ./mnt/workspace/ACE-Step-1.5/acestep/handler.py<lambda>)AceStepHandler.__init__.<locals>.<lambda>   s	      UZr<   ACESTEP_DEBUG_STATSrecordsz.cacheacestepzprogress_estimates.jsong      ?)8modelconfigdevicetorchfloat32dtypevaetext_encodertext_tokenizersilence_latentsample_ratereward_model
batch_size6acestep.core.generation.handler.lyric_alignment_commonr3   dictcustom_layers_configoffload_to_cpuoffload_dit_to_cpucompiledcurrent_offload_costosenvirongetlowergetattrsysstderrdisable_tqdmdebug_stats_last_diffusion_per_step_sec	threadingLock_progress_estimates_lock_progress_estimatespathjoin_get_project_root_progress_estimates_path_load_progress_estimateslast_init_paramsquantizationlora_loadeduse_lora
lora_scale_base_decoder_active_loras_lora_adapter_registry_lora_active_adaptermlx_decoderuse_mlx_ditmlx_dit_compiledmlx_vaeuse_mlx_vae_get_gpu_cfgmlx_vae_chunk_size)selfr3   s     r=   __init__AceStepHandler.__init__g   s   
]]
  !" # ! ! 
 	b$()?$@!#"'$'!JJNN+A2FLLNRff  ]nuvy  wA  wA  CK  MZ  o[  o]  k]::>>*?DJJLPdd=A)(1(8%$-r? (*""$%	)
% 	%%' $ ! !!&(#$(!    %   #/."C"Cr<   )%rp   ro   r`   rr   rq   rd   rc   rh   rO   rU   rD   rV   rR   r_   rE   r^   rH   rj   rl   rn   rs   ru   rv   ry   rC   rT   rS   rk   rN   rM   rL   rJ   rK   rm   rt   rw   rI   N)__name__
__module____qualname____firstlineno____doc__r{   __static_attributes__r;   r<   r=   r.   r.   >   s    N *HDr<   r.   )4r   rW   r\   rX   ra   typingr   rF   warningsacestep.gpu_configr   rx   acestep.core.generation.handlerr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   filterwarningsr.   r;   r<   r=   <module>r      s@   
 
 (/

# $     D& & & & & & & & & &R 	   !qDqDqD qD 	qD
 qD qD qD qD qD qD qD qD qD qD qD  !qD" #qD$ %qD& 'qD( )qD* +qD, -qD. /qD0 1qD2 3qD4 5qD6 7qD8 9qD: ;qD< =qD> ?qD@ AqDB CqDD EqDF  GqDH  IqDJ  KqDr<   