Xi[jSrSSKrSSKrSrSSKJr \(d\R R\5r \R R\R R\ 55r \R R\ S5r \R R\ S5r\R R\ 5(a\"\ 5 \"S\ 35 O4\R R\5(a\"\5 \"S\S35 S rS Hr\R&R)\S5 M! S \R&S 'S SKJr S SKJr S SKJr S SKJr S SKJr SSKJ r J!r! S SK"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r* S SK+J,r, SSjr6S\7S-S\74Sjr8Sr9\:S:Xa\9"5 gg!\a Nf=f!\a \R R\R R\R R\555r-\-\R ;a\R R]S\-5 SSK/Jr SSK0Jr SSK1Jr SSK2Jr SSK3Jr SSKJ r J!r! SSK4J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r* SSK5J,r, Nf=f)z@ ACE-Step V1.5 Pipeline Handler wrapper connecting model and UI NF) load_dotenvz.envz .env.examplezLoaded configuration from z (fallback)T) http_proxy https_proxy HTTP_PROXY HTTPS_PROXY ALL_PROXYffmpegTORCHAUDIO_USE_BACKEND)parse_quantization_arg)AceStepHandler) LLMHandler)DatasetHandler)create_gradio_interface)get_i18navailable_languages_info)get_gpu_configget_gpu_memory_gbresolve_lm_backendprint_gpu_config_infoset_global_gpu_configVRAM_16GB_MIN_GBVRAM_AUTO_OFFLOAD_THRESHOLD_GBis_mps_platform)ensure_lm_modelcU(a'URS5(aSU;a USnUSnO[5n[5n[5n[ UUUUUS9nU$)a Create Gradio demo interface Args: init_params: Dictionary containing initialization parameters and state. If None, service will not be pre-initialized. Keys: 'pre_initialized' (bool), 'checkpoint', 'config_path', 'device', 'init_llm', 'lm_model_path', 'backend', 'use_flash_attention', 'offload_to_cpu', 'offload_dit_to_cpu', 'init_status', 'dit_handler', 'llm_handler' (initialized handlers if pre-initialized), 'language' (UI language code) language: UI language code ('en', 'zh', 'ja', default: 'en') Returns: Gradio Blocks instance pre_initialized dit_handler llm_handler init_paramslanguage)getr rrr)r!r"rrdataset_handlerdemos ;/mnt/workspace/ACE-Step-1.5/acestep/acestep_v15_pipeline.py create_demor'Wsl&  OO- . . [ (!-0 !-0 $&  l $&O #  D Krequested_backendreturnc[X5nU=(d SR5R5nU(aX2:wa[SUSUS35 U$)zKResolve the startup LM backend against hardware compatibility restrictions.zRequested LM backend 'z,' is not supported on this hardware. Using 'z ' instead.)rstriplowerprint)r) gpu_configresolved_backendnormalized_backends r&_resolve_startup_lm_backendr3s_)*;H+1r88:@@B0D $%7$89&'z 3 r(c Z SSKn[5n[U5 URn[ 5nU(+=(a US:=(a U[ :nUR n[SS35 [S5 [S5 [SUSS35 [S UR35 [S URS URS -S 35 [SURS URS -S 35 [SUR35 [SUR35 [SUR35 [SUR=(d S35 [SS35 U(a[SUSSU35 OHU(a[SUSS[ S35 O)US:a[SUSS[ S35 O [S5 [R R#[R R#[R R%[&555n[R R)US5nUR+SS5n[R,"US S!9 [S"U35 [/5 UR1S#UR2S$9nUR5S%[6S&S'S(9 UR5S)S*S+S,9 UR5S-S*S.S,9 UR5S/[8S0S1S(9 [;5n UR5S2[8S3U V s/sHoSPM sn S4S5R)S6U 55-S79 A UR5S8S9/S:S;9 UR5SS?S(9 UR5S@SAS>SBS(9 UR5SC[8SSDS(9 UR5SE[8SSFS(9 UR5SG[8SH/SIQSJS79 UR5SKSLSSMS(9 UR5SN[8SSOS(9 UR5SP[8U/SQQSRUSS3S79 UR5STSUSSVS(9 UR5SWSXUSYU(aSZOS[S\3S(9 U(d UR<OS>n UR5S]S^U S_U S`3S(9 Sn UR>(aRU(dKSan SSK n U RBRE5(a%U RBRGS5upUSb:aScn UR5Se[NU SfU S`3S(9 UR5Sg[8S/ShQSiS79 UR5Sj[6SSkS(9 UR5SlS*SmS,9 UR5Sn[8SSoS(9 UR5Sp[8SSqS(9 UR5Sr[8SSsS(9 URQ5nURR(aS Ul*URVc$[RXR[St5Ul+UR\c$[RXR[Su5Ul.[RXR[Sv5(a$[RXR[Sv5Ul/UR`(a[Sw5 S Ul*URVc%[RXR[SxSy5Ul+UR\c%[RXR[SzS{5Ul.[RXR[S|S}5Ul/[S~URV35 [SUR\35 [cUR^U5Ul/UR`(a[SUR^35 URd(dHUR\(a7SUR\;a'SUs=:aS::aO OS Ul2[SUSS35 UR\(aYSUs=:a [ :aHO OESUR\;a5UR\R+SS5n[SUSSU35 UUl.SnSnSnURT(Ga[S5 [g5n[i5nURVcpURk5nU(a+SU;aSOUSUl+[SURV35 O.[S[lRnS9 [lRp"S5 [R R%[&5n[R R#[R R#U55nURrnUcURuURv5nSnURx(a*URxSH:waURxn[SU35 [SURVSURvS35 [RXR[SS5R{5R}5S;nURUURVURvUUURdURURUS9 unnU(d1[SU3[lRnS9 [lRp"S5 [S5 URc*URUlB[SURS35 SnUR(GaUR\cYUR5nU(a#USUl.[SUR\35 O[S[lRnS9 S>UlBUR(Ga?UR\(Ga-[R R)US5nSnURx(aURxSH:wa URxn[UR\UUS9un n!U (d[SU!3[lRnS9 [SUR\SURvS35 URUUR\UR^URvURdSS9unn#U#(a[S5 USU3- nO#[SU3[lRnS9 USU3- n0SS _SUR`_SUR_SURV_SURv_SUR_SUR\_SUR^_SU_SURd_SUR_SUR_SU_SU_SU_SU_SUR_UUURS.En[S5 [SURS35 UcUURUURS.n[UURS9n$[S5 U$RSSHSS9 [SURSURS35 Sn%UR(a4UR(a#URUR4n%[S5 U/n&URH%n'U'(dM U'U&;dMU&RU'5 M' URR(a[S5 SSKQJRn( U$RURURURURS S S>U%U&S9 U("U$UUURS9 UR(a [S5 [S5 SSKWn)U)RS5 MU$RURURURURS S>S>U%U&S9 gs sn f![Ha!n[JRMSdU5 SnAG NSnAff=f![Ha&n"[SU"3[lRnS9 Sn"A"GNSn"A"ff=f![a [S5 gf=f![HaOn"[SU"3[lRnS9 SSKZn*U*R5 [lRp"S5 Sn"A"gSn"A"ff=f)zMain entry functionrN z<============================================================zGPU Configuration Detected:z GPU Memory: z.2fz GBz Configuration Tier: z Max Duration (with LM): zs (<z min)z Max Duration (without LM): z Max Batch Size (with LM): z Max Batch Size (without LM): z Default LM Init: z Available LM Models: Noneu5Apple Silicon (MPS) detected — unified memory z.1fz#GB, no CPU offload needed, backend=zAuto-enabling CPU offload (GPU zGB < z GB threshold)z%CPU offload disabled by default (GPU zGB >= zNo GPU detected, running on CPUgradio_outputs\/T)exist_okzOutput directory: zGradio Demo for ACE-Step V1.5) descriptionformatter_classz--portiz Port to run the gradio server on)typedefaulthelpz--share store_truezCreate a public link)actionr@z--debugzEnable debug modez --server-namez 127.0.0.1z@Server name (default: 127.0.0.1, use 0.0.0.0 for all interfaces)z --languageenzUI language: z c3T# UHupnUSU3-X#:waSUS3OS-v M g7f)z (r:)N).0codename native_names r& main..sI 0C+D {m$%"&"5QtfA;3@0Cs&()r>r?choicesr@z--allowed-pathappendz6Additional allowed file paths for Gradio (repeatable).)rBr?r@z--service_modec(UR5S;$N)true1yesr.xs r&main..qwwy$88r(Fz`Enable service mode (default: False). When enabled, uses preset models and restricts UI options.z--init_servicec(UR5S;$rPrTrUs r&rWrXrYr(z.Initialize service on startup (default: False)z --checkpointz5Checkpoint file path (optional, for display purposes)z --config_pathz+Main model path (e.g., 'acestep-v15-turbo')z--deviceauto)r[cudampsxpucpuz!Processing device (default: auto)z --init_llmc(UR5S;$rPrTrUs r&rWrX rYr(z5Initialize 5Hz LM (default: auto based on GPU memory)z--lm_model_pathz/5Hz LM model path (e.g., 'acestep-5Hz-lm-0.6B')z --backend)vllmptmlxz5Hz LM backend (default: z2, use 'mlx' for native Apple Silicon acceleration)z--use_flash_attentionc(UR5S;$rPrTrUs r&rWrX3rYr(z*Use flash attention (default: auto-detect)z--offload_to_cpuc(UR5S;$rPrTrUs r&rWrX9rYr(z Offload models to CPU (default: TrueFalsez", auto-detected based on GPU VRAM)z--offload_dit_to_cpuc(UR5S;$rPrTrUs r&rWrXBrYr(z-Offload DiT to CPU after diffusion (default: z", auto-detected based on GPU tier)int8_weight_only w8a8_dynamiczR[parse_args] CUDA capability probe failed while resolving quantization default: {}z--quantizationz\DiT quantization method: int8_weight_only, fp8_weight_only, w8a8_dynamic, or none (default: z--download-source) huggingface modelscoper[zGPreferred model download source (default: auto-detect based on network)z --batch_sizezUDefault batch size for generation (1-8). Defaults to min(2, GPU_max) if not specifiedz --enable-apiz%Enable API endpoints (default: False)z--auth-usernamez"Username for Gradio authenticationz--auth-passwordz"Password for Gradio authenticationz --api-keyz(API key for API endpoints authenticationACESTEP_CONFIG_PATHACESTEP_LM_MODEL_PATHACESTEP_LM_BACKENDz8Service mode enabled - applying preset configurations...SERVICE_MODE_DIT_MODELz(acestep-v15-turbo-fix-inst-shift-dynamicSERVICE_MODE_LM_MODELzacestep-5Hz-lm-1.7B-v4-fixSERVICE_MODE_BACKENDraz DiT model: z LM model: z Backend: 4Bz>Auto-enabling CPU offload (4B LM model requires offloading on z.0fzGB GPU)z1.7Bz&WARNING: 4B LM model is too large for z%GB GPU. Downgrading to 1.7B variant: z)Initializing service from command line...zacestep-v15-turbozAuto-selected config_path: z>Error: No available models found. Please specify --config_path)filer z!Using preferred download source: zInitializing DiT model: z on z...ACESTEP_COMPILE_MODELr,>rRyonrSrQ) project_root config_pathdeviceuse_flash_attention compile_modeloffload_to_cpuoffload_dit_to_cpu quantization prefer_sourcezError initializing DiT model: z"DiT model initialized successfullyzAuto-setting init_llm to z based on GPU configurationzUsing default LM model: z;Warning: No LM models available, skipping LM initialization checkpoints) model_namecheckpoints_dirrz#Warning: LM model download failed: z&Warning: Failed to download LM model: zInitializing 5Hz LM: )checkpoint_dir lm_model_pathbackendr|rdtypez5Hz LM initialized successfullyz'Warning: 5Hz LM initialization failed: r service_mode checkpointr{r|init_llmrrr}rrr init_statusenable_generaterrr")r0 output_dirdefault_batch_sizez.Service initialization completed successfully!z)Creating Gradio interface with language: )r0r"rrr z(Enabling queue for multi-user support...)max_sizestatus_update_ratedefault_concurrency_limitzLaunching server on :zAuthentication enabledzEnabling API endpoints...)setup_api_routes) server_name server_portsharedebug show_errorprevent_thread_lock inbrowserauth allowed_paths)api_keyzAPI authentication enabledzoAPI endpoints enabled: /health, /v1/models, /release_task, /query_result, /create_random_sample, /format_lyricsz Shutting down...zError launching Gradio: )\argparserr gpu_memory_gbrrrecommended_backendr/tiermax_duration_with_lmmax_duration_without_lmmax_batch_size_with_lmmax_batch_size_without_lminit_lm_defaultavailable_lm_modelsospathdirnameabspath__file__joinreplacemakedirsrArgumentParserRawTextHelpFormatter add_argumentintstrroffload_dit_to_cpu_defaultquantization_defaulttorchr\ is_availableget_device_capability Exceptionloggerwarningr parse_args enable_api init_servicer{environr#rrrr3rr r get_available_acestep_v15_modelssysstderrexitr}is_flash_attention_availabler|download_sourcer-r.initialize_servicerrrget_available_5hz_lm_modelsr initializerr" batch_sizer'queuerport auth_username auth_password allowed_pathrN acestep.ui.gradio.api.api_routesrlaunchrrrtimesleepKeyboardInterrupt traceback print_exc)+rr0r_is_mac auto_offload_default_backendrzrparseravailable_languagesr"_default_offload_dit_default_quantizationrmajor_excargsfallbackr!rravailable_models current_filer}rr~rr lm_statusrrdl_okdl_msge lm_successr%rrprrrs+ r&mainrs| !J*%,,MG ; A  ; : : "55 Bxj/ '( XJ N=-S 12 ":??"3 45 $Z%D%D$ESIhIhlnInHootu ' (J(J'K3zOqOquwOwNxx}~ ()J)J(K LM +J,P,P+Q RS  : :; <= #J$B$B$Lf#M NO XJb/ CMRUCVVy{KzL M   -mC-@FdEeer s    3M#3FfMkLlly z  /077??277??277??83L#MNLl,<=J##D#.JKK T* zl +, J  $ $3 55%F sD/Q  ,=ST  ,=PQ   O 34  -@A-@!-@A  ++ 0C       E   8 o   8 =     D     :    5 0   8 D     >     %()9(::l m   8 9    8/,G/TTv w 6= --%  8$<=Q19,:)   #% .//Q S    5 V    d   4   1     1     7     D      #!zz~~.CDD     %!#0G!HD  ::>>. / /::>>*>?DL  HI     #!zz~~(*T D     %!#')E"D zz~~&*+   4#5#5$$BTBT:T } " ""&D  PQ^_bPccjk  a-P2PP 4%% %))11$?H 8s8KL008z; "*D ~        = >)*K$,K'#.#O#O#Q #/2BB,-a0$ 78H8H7IJKX ZZHHQK77??84L77??277??<+HIL#'":": "*&1&N&NKK'# !M##(<(<(F $ 4 4 9-IJ ,T-=-=,>d4;;-sS TJJNN'egeeg!@AM,7+I+I) ,,{{$7+#22#'#:#:!..+,J , (K#6{mD3::V 6 8}}$ * : : / >YZI}}}%%-*5*Q*Q*S'*-@-C* 89K9K8LMNY!$). ===T%7%7%7%'WW\\, %NN%)M++0D0D0N(,(<(< (7'+'9'9,:*7) v %!"EfX N%(ZZ/0B0B/C4 }TWX-8,B,B'5&*&8&8 $ #{{'+':':" -C-)Iz" ?A#I;'77 Ei[Q!$$I;'77 !4 1 1doot//  $++  DMM  !3!34<<&':!$"5"5%d&=&= 1 1{"?{ {!"DMM#$)(&*oo)K. B C 9$--LM  ( MM(&*oo K{T]]K 89 %&'  $T%5%5$6a {#FG   $"4"4&&(:(:;D * +# ""AqQm+$$Q'# ?? - . I KK ,, IIjjjj$(+   T; T\\ R||23 B   ,JJqM KK ,, IIjjjj$)+  SBN  NNd   H%DQCH!$v% ,*+ ,  (,3::>  s}A}"N5:~/I ?B08~6? ~};;~ ~3 ~.(.~336   A@*AA@%@%A@*__main__)NrC);__doc__rr _env_loadeddotenvrrrr _current_filer _project_rootr _env_path_env_example_pathexistsr/ ImportError proxy_varrpopcli_argsr handlerr llm_inferencerr$r ui.gradioracestep.ui.gradio.i18nrrr0rrrrrrrrmodel_downloaderrrzinsertacestep.cli_argsacestep.handleracestep.llm_inferenceacestep.dataset_handleracestep.ui.gradioacestep.gpu_configacestep.model_downloaderr'rr3r__name__rFr(r&r s     " 1  (FG GGLL7 GGLLG 77>>) $ $  " .yk: ; WW^^- . . ) * ./@.AM N IJJNN9d#(0 #$(90')/2I   22(V 3: c qh zFS  H977??277??277??83L#MNL388# <(7.069I   9+9s&C>F$AF0$F-,F-0B?I21I2