#!/bin/bash
# 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"