SonicMaster: Towards Controllable All-in-One Music Restoration and Mastering
Paper • 2508.03448 • Published • 6
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("AEmotionStudio/sonicmaster-models", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Self-contained inference bundle for the MAESTRO desktop app.
sonicmaster/model.safetensors — diffusion checkpoint (source: amaai-lab/SonicMaster)sonicmaster/tangoflux_config.yaml — model architecture configt5/ — google/flan-t5-large text encoder + tokenizer (Apache-2.0)vae/ — Oobleck VAE from stabilityai/stable-audio-open-1.0Paper: Melechovsky, Mehrish, Herremans. "SonicMaster: Towards Controllable All-in-One Music Restoration and Mastering." ICML 2026 (arXiv:2508.03448).