TowstyandCursor
de433c40d3
Keep mmgp profile 2 (no quanto), node-legal token floor, two Stage-A segments, and document why profiles 3-5 blow up under InferenceMode. Co-authored-by: Cursor <cursoragent@cursor.com>
76 lines
1.8 KiB
JSON
76 lines
1.8 KiB
JSON
{
|
|
"15": {
|
|
"inputs": {
|
|
"stage_A_repo": "m-a-p/YuE-s1-7B-anneal-en-cot",
|
|
"xcodec_ckpt": "ckpt_00360000.pth",
|
|
"quantization_model": "fp16",
|
|
"use_mmgp": true,
|
|
"stage1_cache_size": 8192,
|
|
"exllamav2_cache_mode": "FP16",
|
|
"mmgp_profile": 2
|
|
},
|
|
"class_type": "YUE_Stage_A_Loader",
|
|
"_meta": {
|
|
"title": "YuE Stage A Loader"
|
|
}
|
|
},
|
|
"8": {
|
|
"inputs": {
|
|
"genres_prompt": "inspiring female uplifting pop airy vocal electronic bright",
|
|
"lyrics_prompt": "[verse]\n\n[chorus]\n",
|
|
"seed": 0,
|
|
"run_n_segment": 2,
|
|
"repetition_penalty": 1.1,
|
|
"prompt_start_time": 0,
|
|
"prompt_end_time": 45,
|
|
"max_new_tokens": 2944,
|
|
"use_dual_tracks_prompt": false,
|
|
"use_audio_prompt": false,
|
|
"offload_model": true,
|
|
"stage1_no_guidance": true,
|
|
"model": ["15", 0]
|
|
},
|
|
"class_type": "YUE_Stage_A_Sampler",
|
|
"_meta": {
|
|
"title": "YuE Stage A"
|
|
}
|
|
},
|
|
"16": {
|
|
"inputs": {
|
|
"stage_B_repo": "m-a-p/YuE-s2-1B-general",
|
|
"stage2_cache_size": 4096,
|
|
"stage2_batch_size": 1,
|
|
"exllamav2_cache_mode": "FP16",
|
|
"use_mmgp": true,
|
|
"info": ["8", 1]
|
|
},
|
|
"class_type": "YUE_Stage_B_Loader",
|
|
"_meta": {
|
|
"title": "YuE Stage B Loader"
|
|
}
|
|
},
|
|
"9": {
|
|
"inputs": {
|
|
"vocal_decoder_ckpt": "decoder_131000.pth",
|
|
"inst_decoder_ckpt": "decoder_151000.pth",
|
|
"rescale": true,
|
|
"stage1_set": ["8", 0],
|
|
"model": ["16", 0]
|
|
},
|
|
"class_type": "YUE_Stage_B_Sampler",
|
|
"_meta": {
|
|
"title": "YuE Stage B"
|
|
}
|
|
},
|
|
"19": {
|
|
"inputs": {
|
|
"filename_prefix": "audio/AIGen",
|
|
"audio": ["9", 0]
|
|
},
|
|
"class_type": "SaveAudio",
|
|
"_meta": {
|
|
"title": "Save Audio"
|
|
}
|
|
}
|
|
}
|