| # usage: yue_sing.sh <lyrics_txt> <genre_txt> <out_dir> <segments> <seed> | |
| cd /marimo/ai_studio/vendor/YuE/inference || exit 1 | |
| export PYTHONPATH=/marimo/ai_studio/vendor/YuE/inference | |
| export PYTHONSAFEPATH= | |
| export YUES_NO_COMPILE=1 | |
| exec /marimo/ai_studio/envs/music/bin/python infer.py \ | |
| --stage1_model m-a-p/YuE-s1-7B-anneal-en-cot \ | |
| --stage2_model m-a-p/YuE-s2-1B-general \ | |
| --genre_txt "$2" --lyrics_txt "$1" \ | |
| --run_n_segments "$4" --stage2_batch_size 4 \ | |
| --output_dir "$3" --cuda_idx 0 --seed "$5" | |