Skip to content

Conversation

@kmraven
Copy link

@kmraven kmraven commented Oct 28, 2025

This PR fixes a runtime error in generate_diffusion_uncond caused by passing an unexpected mask argument to sample_k.

In stable_audio_tools/inference/generation.py, line ~78, the generate_diffusion_uncond calls:

sampled = sample_k(model.model, noise, init_audio, mask, steps, **sampler_kwargs, device=device)

However, sample_k (defined in stable_audio_tools/inference/sampling.py) expects the 4th positional argument to be steps and does not accept a mask parameter at all. Other call sites use sample_k without mask.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant