Nodes/TTS Audio Suite/⚙️ Step Audio EditX Engine
ComfyUI Node

⚙️ Step Audio EditX Engine

The loader that powers Step's TTS and its audio editor

By diodiogod·Created 12 months ago·Updated 18 days ago· 1,098
⚙️ Step Audio EditX Engine
    • TTS_engine
    model_pathStep-Audio-EditX
    deviceauto
    torch_dtypebfloat16
    quantizationnone
    temperature0.7
    do_sampletrue
    max_new_tokens1024
    runtime_mode⚠️ Shared Runtime

    This is the loader node. On its own it doesn't make a sound - it loads the Step Audio EditX model into memory, sets how it runs, and hands out a TTS_ENGINE that the actual working nodes plug into. In TTS Audio Suite the pattern is always the same: an Engine node configures a model, and a downstream node (TTS Text, TTS SRT, or the Audio Editor) does the work. This is the Engine half for Step.

    Step Audio EditX is a two-in-one deal. Feed this engine into a TTS node and it does zero-shot voice cloning. Feed it into the Step Audio EditX - Audio Editor node and it does the emotion/style/paralinguistic second-pass editing that's the model's real party trick. Same engine, two jobs - you pick by what you wire it into.

    How it works

    You set the model, the device, and the precision; the node loads a ~7GB, 3B-parameter Step model and keeps it resident. The TTS_engine output is just a handle to that loaded model. Downstream nodes read it and run inference against it, so you only load once even if several nodes use it.

    The inputs and outputs that matter

    Honestly, most beginners can leave this node on defaults. The few worth knowing:

    • runtime_mode - Shared Runtime, Dedicated Runtime, or Main Environment. This is the suite's isolation system for engines with fussy dependencies. Step is a modern engine that's happy in Main Environment, but if you hit dependency conflicts, the shared/dedicated modes give it its own sandbox instead of forcing your whole ComfyUI backward.
    • quantization - none, int4, or int8. If 7GB of model is straining your VRAM, int8 (and especially int4) shrink the footprint. Expect a small quality cost; often worth it to make it fit.
    • torch_dtype - leave it on bfloat16 unless your GPU is old enough to prefer float16.
    • temperature (default 0.7), do_sample, max_new_tokens (1024) - generation randomness and length. Higher temperature is more varied, lower is more deterministic. Fine at defaults to start.
    • device - auto picks your GPU. Leave it.

    The single output is TTS_engine (TTS_ENGINE). Run it into a TTS Text/SRT node for cloning, or into the Audio Editor for editing.

    How to install it

    It ships in TTS Audio Suite, so install the pack. Via ComfyUI Manager, search TTS Audio Suite, install, restart - the Manager runs the pack's install.py and handles the dependency mess. Manual:

    cd ComfyUI/custom_nodes
    git clone https://github.com/diodiogod/TTS-Audio-Suite.git
    cd TTS-Audio-Suite
    python install.py
    

    then restart. On Linux install portaudio19-dev and libsamplerate0-dev first. The Step model auto-downloads to ComfyUI/models/TTS/step_audio_editx/ on first run, so budget time (and disk) for that ~7GB pull the first time.

    Common issues & troubleshooting

    Out-of-memory on load. 7GB plus overhead is a lot on an 8–12GB card, especially with other models loaded. Drop quantization to int8 or int4, and consider running Step on its own rather than alongside a big image model in the same graph.

    Dependency conflicts on install. Step is meant to be Main-Environment friendly, but if the pack's install trips over something, that's what the runtime_mode shared/dedicated options exist for - set isolation on the engine instead of downgrading your base ComfyUI.

    First run hangs for ages. It's downloading the model, not frozen. Watch the console for the download progress; subsequent runs load from cache and are quick.

    It "works" but the voice is bland. That's the model's character, not a config bug - Step's cloning is decent but a bit neutral. Its strength is the editing pass; lean on the Audio Editor node for expressiveness rather than expecting drama from the raw clone.

    CategoryTTS Audio Suite/⚙️ Engines

    Inputs (8)

    NameTypeDefaultDescription
    model_pathCOMBOStep-Audio-EditXStep Audio EditX model selection: • local:ModelName: Use locally installed model (respects extra_model_paths.yaml) • ModelName: Auto-download model if not found locally • Downloads respect extra_model_paths.yaml configuration
    deviceCOMBOautoDevice to run Step Audio EditX on: • auto: Select the best available device • cuda: NVIDIA GPU; this is a heavy model and 12GB+ VRAM is recommended • cpu: Extremely slow and intended only as a fallback
    torch_dtypeCOMBObfloat16Model precision: • bfloat16: Recommended on supported NVIDIA GPUs • float16: Compatibility alternative with similar memory use • float32: Much higher memory use; generally not recommended • auto: Selects bfloat16 when supported, otherwise float16
    quantizationCOMBOnoneLLM weight quantization: • none: Best quality and compatibility • int8: Reduces LLM weight memory • int4: Stronger reduction with a larger quality/compatibility tradeoff The audio tokenizer and vocoder are not quantized, so total VRAM does not shrink proportionally.
    temperatureFLOAT0.70.1–2Sampling temperature for generation. • Lower (0.3-0.5): More consistent, predictable output • Default (0.7): Balanced creativity and stability • Higher (1.0+): More varied, potentially unstable
    do_sampleBOOLEANtrueEnable sampling for generation. • True: Uses temperature for varied output (recommended) • False: Greedy decoding can fail to emit an end token and is not recommended
    max_new_tokensINT1024128–8192Maximum generated audio tokens (safety ceiling): • 512: roughly 12s • 1024: roughly 25s (recommended) • Higher values allow longer output but increase runtime and KV-cache memory.
    runtime_modeCOMBO⚠️ Shared RuntimePython runtime used by Step Audio EditX: • Shared Runtime: Recommended. Uses the shared Transformers 4 runtime and was verified with Step EditX. • Dedicated Runtime: Uses a separate Step EditX Transformers 4 environment. Choose this only to isolate dependency conflicts. • Main Environment: Uses ComfyUI's packages. Transformers 5 currently produces invalid Step audio tokens and is not recommended.

    Outputs (1)

    NameTypeDescription
    TTS_engineTTS_ENGINE