From fee84bc46967cb46ebb4036a79072bb32202db56 Mon Sep 17 00:00:00 2001
From: Matthew Rice <mattricesound@gmail.com>
Date: Fri, 5 Jun 2026 15:19:41 -0400
Subject: [PATCH] Fix cli so that it uses the correct sample size

---
 stable_audio_3/cli.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/stable_audio_3/cli.py b/stable_audio_3/cli.py
index 7b43ac0..192e462 100644
--- a/stable_audio_3/cli.py
+++ b/stable_audio_3/cli.py
@@ -267,6 +267,7 @@ def main():
             args.inpaint_ends[0] if len(args.inpaint_ends) == 1 else args.inpaint_ends
         )
 
+    sample_size = model.model_config["sample_size"]
     # --- Generate ---
     print("Generating…")
     audio = model.generate(
@@ -277,6 +278,7 @@ def main():
         cfg_scale=args.cfg_scale,
         seed=args.seed,
         batch_size=batch_size,
+        sample_size=sample_size,
         init_audio=init_audio,
         init_noise_level=args.init_noise_level,
         inpaint_audio=inpaint_audio,