From d467f040cb5b2ee5324bc922124aa59c9e66bd22 Mon Sep 17 00:00:00 2001 From: Ryota Kimura <128337097+kmraven@users.noreply.github.com> Date: Tue, 28 Oct 2025 17:48:57 +0900 Subject: [PATCH] Fix generate_diffusion_uncond: remove unsupported mask arg in sample_k call --- stable_audio_tools/inference/generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_audio_tools/inference/generation.py b/stable_audio_tools/inference/generation.py index 35e63073..2228f4f7 100644 --- a/stable_audio_tools/inference/generation.py +++ b/stable_audio_tools/inference/generation.py @@ -75,7 +75,7 @@ def generate_diffusion_uncond( if diff_objective == "v": # k-diffusion denoising process go! - sampled = sample_k(model.model, noise, init_audio, mask, steps, **sampler_kwargs, device=device) + sampled = sample_k(model.model, noise, init_audio, steps, **sampler_kwargs, device=device) elif diff_objective in ["rectified_flow", "rf_denoiser"]: sampled = sample_rf(model.model, noise, init_data=init_audio, steps=steps, **sampler_kwargs, device=device)