add eval lakh ids

This commit is contained in:
slSeanWU
2025-11-05 11:02:55 -05:00
parent bbe8626b7d
commit 94f40b290f
3 changed files with 904 additions and 2 deletions
+4 -2
View File
@@ -81,11 +81,11 @@ Outputs 4 MIDIs (and synthesized MP3s) conditioned on the same prompt by default
```bash
python generate_vllm.py \
--prompts_file some_example_prompts.txt \
--prompts_file assets/example_prompts.txt \
--fp8 \
--no-synthesize
```
- `some_example_prompts.txt` should contain one prompt per line.
- [`assets/example_prompts.txt`](assets/example_prompts.txt) contains 4 example prompts (one per line).
- `--fp8` performs FP8 quantization for faster inference.
- `--no-synthesize` skips audio synthesis (outputs MIDI only).
@@ -184,6 +184,8 @@ We provide high-level guidance for researchers interested in training their own
3. **Collect text prompts**: Obtain text descriptions for your MIDI files (e.g., [MidiCaps](https://huggingface.co/datasets/amaai-lab/MidiCaps) in our use case)
4. **Match text-MIDI examples**: Ensure you can map each text prompt to its corresponding MIDI file
**Note**: The 896 LakhMIDI IDs used for evaluation in our paper are available in [`assets/evaluation_set_lakh_ids.txt`](assets/evaluation_set_lakh_ids.txt).
</details>
<details>