towsty/aigen
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add YuE2 to the music engine picker and generate API.

Browse Source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Towsty
2026-09-14 18:26:55 -05:00
co-authored by Cursor
parent 420dd1279c
commit 98f306a88a
4 changed files with 68 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/music.ts
+5 -2
View File
@@ -33,22 +33,25 @@ export const DEFAULT_MUSIC_LYRICS = `[Verse 1]
[Outro]
`
export type MusicEngine = 'ace-step' | 'ace-step-1.5' | 'yue'
export type MusicEngine = 'ace-step' | 'ace-step-1.5' | 'yue' | 'yue2'
export function parseMusicEngine(raw: unknown): MusicEngine {
const value = String(raw || '').trim().toLowerCase()
if (value === 'ace-step-1.5' || value === 'ace15' || value === 'v1.5') return 'ace-step-1.5'
if (value === 'yue2' || value === 'yue-2') return 'yue2'
if (value === 'yue' || value === 'yue-s1' || value === 'lyrics2song') return 'yue'
return 'ace-step'
}
export function isAceMusicEngine(engine: MusicEngine | string | undefined) {
return parseMusicEngine(engine) !== 'yue'
const value = parseMusicEngine(engine)
return value !== 'yue' && value !== 'yue2'
}
export function musicEngineLabel(engine: MusicEngine | string | undefined) {
const value = parseMusicEngine(engine)
if (value === 'ace-step-1.5') return 'ACE-Step 1.5'
if (value === 'yue2') return 'YuE2'
if (value === 'yue') return 'YuE'
return 'ACE-Step'
}