Keep YuE2 jobs live on the bench and wire target length to semantic tokens.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-15 20:53:07 -05:00
parent 2ebc3066e6
commit 5694104226
13 changed files with 92 additions and 29 deletions
@@ -1,6 +1,13 @@
-export const MUSIC_DURATION_MIN = 30
+export const MUSIC_DURATION_MIN = 15
 export const MUSIC_DURATION_MAX = 150
 export const MUSIC_DURATION_DEFAULT = 60
+/** YuE2 VAE downsampling: 48000 / 1920 = 25 semantic tokens per second of audio. */
+export const YUE2_TOKENS_PER_SECOND = 25
+
+export function yue2MaxTokens(durationSec: number) {
+  const seconds = clampMusicDuration(durationSec)
+  return Math.max(200, Math.min(9000, seconds * YUE2_TOKENS_PER_SECOND))
+}
 export const MUSIC_STEPS_MIN = 8
 export const MUSIC_STEPS_MAX = 100
 export const MUSIC_STEPS_DEFAULT = 50