nidZddlZddlZdejd<ddlZddlmZddlZddlZddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ejddGd d geeeeeee eeeeeeeeeeeee e!e"e#e$e%e&e'e(e)e*e+e,e-e.e/e0e1Z3y) zm Business Logic Handler Encapsulates all data processing and business logic as a bridge between model and UI NfalseTOKENIZERS_PARALLELISM)Optional)get_global_gpu_config)%AudioCodesMixinBatchPrepMixinConditioningBatchMixinConditioningEmbedMixinConditioningMaskMixinConditioningTargetMixinConditioningTextMixinDiffusionMixinGenerateMusicDecodeMixinGenerateMusicExecuteMixinGenerateMusicMixinGenerateMusicPayloadMixinGenerateMusicRequestMixinInitServiceMixin IoAudioMixinLyricScoreMixinLyricTimestampMixinLoraManagerMixinMemoryUtilsMixin MetadataMixinMlxDitInitMixinMlxVaeDecodeNativeMixinMlxVaeEncodeNativeMixinMlxVaeInitMixin PaddingMixin ProgressMixin PromptMixinServiceGenerateMixinTrainingPresetMixinTaskUtilsMixinVaeDecodeChunksMixinVaeDecodeMixinVaeEncodeChunksMixinVaeEncodeMixinServiceGenerateRequestMixinServiceGenerateExecuteMixinServiceGenerateOutputsMixinignoreceZdZdZdZy)AceStepHandlerzACE-Step Business Logic Handlerc d|_d|_d|_tj|_d|_d|_d|_d|_ d|_ d|_ d|_ ddl m}t||_d|_d|_d|_d|_t*j,j/d d j1d vxs"t3t4j6d d  |_t*j,j/dd j1d v|_d|_t?j@|_!dgi|_"t*jFjI|jKddd|_&|jOd|_(d|_)d|_*d|_+d|_,d|_-i|_.i|_/d|_0d|_1d|_2d|_3d|_4d|_5tmjn|_7y)zFInitialize runtime model handles, feature flags, and generation state.Ncpuir)_DEFAULT_LAYERS_CONFIGFgACESTEP_DISABLE_TQDM)1trueyesisattycy)NFr:F/mnt/workspace/acestep.cpp/tests/../../ACE-Step-1.5/acestep/handler.pyz)AceStepHandler.__init__..s UZr;ACESTEP_DEBUG_STATSrecordsz.cacheacestepzprogress_estimates.jsong?)8modelconfigdevicetorchfloat32dtypevae text_encodertext_tokenizersilence_latent sample_rate reward_model batch_size6acestep.core.generation.handler.lyric_alignment_commonr2dictcustom_layers_configoffload_to_cpuoffload_dit_to_cpucompiledcurrent_offload_costosenvirongetlowergetattrsysstderr disable_tqdm debug_stats_last_diffusion_per_step_sec threadingLock_progress_estimates_lock_progress_estimatespathjoin_get_project_root_progress_estimates_path_load_progress_estimateslast_init_params quantization lora_loadeduse_lora lora_scale _base_decoder _active_loras_lora_adapter_registry_lora_active_adapter mlx_decoder use_mlx_ditmlx_dit_compiledmlx_vae use_mlx_vae _get_gpu_cfgmlx_vae_chunk_size)selfr2s r<__init__zAceStepHandler.__init__gs    ]] !"#!! b$()?$@!#"' $'!JJNN+A2FLLNRff]o[nuvywAwACKMZo[o]k]::>>*?DJJLPdd=A)(1(8%$-r? (*  " " $   % ) % %%' $!! !&(#$(!   %  #/."C"Cr;N)__name__ __module__ __qualname____doc__ryr:r;r<r.r.>sN*HDr;r.)4r}rUrZrVr_typingrrDwarningsacestep.gpu_configrrvacestep.core.generation.handlerrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+filterwarningsr.r:r;r<rsB (/ #$ D&&&&&&&&&&R!qDqDqDqD qD  qD  qDqDqDqDqDqDqDqDqDqD !qD"#qD$%qD&'qD()qD*+qD,-qD./qD01qD23qD45qD67qD89qD:;qD<=qD>?qD@AqDBCqDDEqDF GqDH IqDJ KqDr;