Return a job record by ID if present.

Convenience function for translation Args: key: Translation key **kwargs: Optional format parameters Returns: T

Get value

Return kwargs for direct assertion in tests.

Accept join calls without background threading.

Return self for chained ``to(...)`` transitions in tests.

Time a code section with GPU synchronization.

Check if key exists

Return configured record by ID, or ``None`` when missing.

Mirror torch module API used by score mixin.

Return self to emulate in-place module migration.

Get parameter as float with fallback default.

Return deterministic JSON payload for this test request.

Return placeholder params tree.

Ask for a single value with an optional default. Args: label: Prompt text. default: Default value (None = required). requ

Return deterministic latent tensor (2x temporal downsample).

Decrypt and return a stored secret value.

Validate and normalise a user-provided path. The returned path is guaranteed to live under *base* (or the global ``_SAFE_ROOT`` when *base* i

Run model forward and sampling. For CFG sequences, batch is structured as: [cond_seq1, cond_seq2, ..., uncond_seq1, uncond_seq2, ...]

Decode latents -> audio. Args: latents_nlc: [B, L_latent, C_latent] in NLC format. Returns: audio: [B, L_aud

Return True when Rich output should be used.

Return iterable form keys.

Return the 46-output mode-switch update tuple for generation UI.

Get parameter as string with fallback default.

Apply whitelist constraint inplace: only allow specified tokens, block all others. This is more efficient than creating a ma

Load ``batch_management.py`` with dependency stubs and trackers. Args: is_windows: Simulate Windows platform flag in ``generation_info``

Print via Rich console if available, else plain print.

Return ``(dwt, iwt)`` for the requested device/dtype/wavelet. Returns ``None`` when ``pytorch_wavelets`` is missing — callers must tr

Capture latest non-empty stripped message.

Encrypt and store a secret value.

Extract model name from config path.

Print a section header.

Read boolean environment variable with legacy truthy semantics.

Get the checkpoints directory path. Resolution order: 1. *custom_dir* argument (passed programmatically) 2. ``ACESTEP_CHECKPOINTS_DIR`` e

Get list of matching keys Note: Simplified implementation, only supports prefix and full matching

Get parameter as bool with fallback default.

Return a no-op context manager to mimic handler model context.

Apply the configured DCW correction. Args: x_next: Latent produced by the sampler step, shape ``[B, T, C]``. denoised

Print debug message if NANOVLLM_DEBUG is enabled

Convert Markdown to HTML with basic formatting. Handles headings, bold, italic, code blocks, lists, blockquotes, and paragraphs. Intentional

Build a minimally valid ``generate_with_progress`` result tuple.

Create an inline (?) help button that opens a modal with documentation. The entire widget (button + hidden modal) is a single ``gr.HTML`` com

No-op cache clear for test host.

Emit an END log with elapsed ms if enabled and start_ts is present.

Get the global GPU configuration, initializing if necessary. Respects the ``ACESTEP_SAVE_MEMORY`` environment variable: when set to ``"1"`` o

Save audio data to file Args: audio_data: Audio data, torch.Tensor [channels, samples] or numpy.ndarray

Mark event as set.

Escape Rich markup characters in user-provided text for safe display. Replaces ``[`` with ``\\[`` so that paths and other user input containing

Check if *token* appears as a delimited word in *path*. Matches when *token* is bounded by start/end of string or a common path delimiter (``

Print a message using Rich if available, plain otherwise.

Apply linear fade in and/or fade out to audio data. Args: audio_data: Audio data as torch.Tensor [channels, samples] or numpy.ndarray.

Return next payload chunk, emulating Starlette UploadFile reads.

Ask for a yes/no boolean value. Raises: GoBack: When ``allow_back`` is True and user types 'b'/'back'.

Return self to emulate in-place dtype cast.

Get current CUDA memory statistics in GB.

Display a numbered menu and return the chosen key. Args: title: Prompt text. options: List of ``(key, label)`` tuples. de

Validate request auth from body ``ai_token`` or Authorization header. Args: body: Parsed request payload dictionary. authorizatio

Build complete kwargs for ``generate_with_batch_management``.

Format seconds to human readable string (e.g. ``2m 30s``).

Return a stable dtype for VAE-related tests.

Return tagged decode output.

Return ``("ok", True)`` to emulate successful DiT initialization.

Run the complete MLX diffusion loop with optional CFG guidance. This is the core generation function. It accepts numpy arrays (converted fro

Return stable dictionary representation used by route persistence logic.

Select MLX VAE decode chunk size based on available unified memory. The ``ACESTEP_MLX_VAE_CHUNK`` environment variable takes highest priority

Check if tensor is on the target device (handles cuda vs cuda:0 comparison).

Build a mock decoder that returns input-dependent velocity predictions. Uses a simple function of the input so that Heun (which evaluates at a

Capture normalize args and return deterministic normalized payload.

Load lyric alignment layers config from model config if available. Called after ``self.config`` is set during model initialization. T

Build GenerationParams and GenerationConfig from request and prepared LLM inputs. Args: req: Request object for music generation.

Return a start timestamp (perf counter) if enabled, otherwise None.

Ensure model exists locally, downloading from configured source if missing.

Generate music using ACE-Step model with optional LM reasoning. Args: dit_handler: Initialized DiT model handler (AceStepHandler inst

Get GPU configuration based on detected or provided GPU memory. On macOS with MPS (Apple Silicon), several overrides are applied automat

Capture initialize_service arguments and return a configured test result.

Return ``(src_latents, enc_hs, enc_am, ctx, attn)`` with fixed seeds.

Synchronize GPU operations for accurate timing.

Validate user-supplied audio path and block unsafe filesystem traversal. Args: path: User-supplied path value from request payload.

Call _build_chunk_masks_and_src_latents with sensible defaults.

Create a route-wired app and return ``TestClient`` for HTTP assertions.