Software & Data Downloads — TUSS

Task-Aware Unified Source Separation for separating audio sources using a variable number of learnable prompts.

PyTorch code for task-aware unified source separation (TUSS), a separation model that uses a variable number of learnable prompts to specify which source to separate, and changes its behavior depending on the given prompts, enabling it to handle all the major separation tasks.