{
  "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": 16384,
      "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": 30,
      "max_new_tokens": 3000,
      "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": 8192,
      "stage2_batch_size": 2,
      "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"
    }
  }
}