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

Replace Comfy YuE with standalone YuEGP profile 1

Browse Source
This commit is contained in:
Towsty
2026-09-07 20:24:47 -05:00
parent eeff12cdab
commit c2b6032e49
33 changed files with 976 additions and 294 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/download_music_models.ps1
+1 -25
View File
@@ -177,7 +177,6 @@ $vae = Join-Path $root 'models\vae'
$ttsAce = Join-Path $root 'models\TTS\ACE-Step-v1-3.5B'
$customAce = Join-Path $root 'custom_nodes\ComfyUI-ACE-Step\models'
$yueRoot = Join-Path $root 'models\YuE'
$yueCustom = Join-Path $root 'custom_nodes\ComfyUI-YuE\models'
Write-Step "ComfyUI root"
Write-Host " $root"
@@ -283,30 +282,7 @@ if (-not $SkipAceStep15) {
}
}
if (-not $SkipYue) {
$s1 = Join-Path $yueRoot 'YuE-s1-7B-anneal-en-cot'
$s2 = Join-Path $yueRoot 'YuE-s2-1B-general'
$codec = Join-Path $yueRoot 'xcodec_mini_infer'
Pull-IfMissing -Label 'YuE stage-1 (s1 7B anneal EN CoT)' -Target $s1 -MinBytes 1GB -Download {
Invoke-HfDownload -Cli $cli -Repo 'm-a-p/YuE-s1-7B-anneal-en-cot' -LocalDir $s1 -Include @('*.safetensors', '*.bin', '*.json', '*.model', '*.txt')
}
Pull-IfMissing -Label 'YuE stage-2 (s2 1B general)' -Target $s2 -MinBytes 200MB -Download {
Invoke-HfDownload -Cli $cli -Repo 'm-a-p/YuE-s2-1B-general' -LocalDir $s2 -Include @('*.safetensors', '*.bin', '*.json', '*.model', '*.txt')
}
Pull-IfMissing -Label 'YuE vocoder (xcodec_mini_infer)' -Target $codec -MinBytes 20MB -Download {
Invoke-HfDownload -Cli $cli -Repo 'm-a-p/xcodec_mini_infer' -LocalDir $codec -Include @('*.safetensors', '*.pth', '*.pt', '*.json', '*.yaml', '*.txt')
}
if (Test-Path (Split-Path $yueCustom -Parent)) {
Pull-IfMissing -Label 'YuE custom-node models copy' -Target $yueCustom -MinBytes 1GB -Download {
Ensure-Dir $yueCustom
Copy-Item -Recurse -Force (Join-Path $yueRoot '*') $yueCustom
}
} else {
Write-Skip "ComfyUI-YuE custom node not installed — YuE weights stay in models\YuE"
}
}
if (-not $SkipYue) { Write-Host 'YuE now uses standalone YuEGP. Run scripts/setup-yuegp.ps1.' }
$after = Get-DirBytes (Join-Path $root 'models')
$delta = [Math]::Max(0, $after - $before)