add loudnorm; default model to hf hub

This commit is contained in:
slSeanWU
2025-10-30 14:07:49 -04:00
parent 2b9d0a8227
commit cd7b889626
3 changed files with 38 additions and 8 deletions
+3 -3
View File
@@ -2,6 +2,7 @@
"""
This script generates MIDI files from text prompts using the MIDI-LLM model with vLLM backend.
vLLM provides faster inference compared to standard HuggingFace model.generate() mixin.
Caveat is that initialization + compilation takes more time, so best used for inference with many prompts.
"""
import json
@@ -235,8 +236,8 @@ Examples:
parser.add_argument(
"--model",
type=str,
required=True,
help="Path to MIDI-LLM model checkpoint"
default="slseanwu/MIDI-LLM_Llama-3.2-1B",
help="Path to MIDI-LLM model checkpoint, can be HuggingFace model ID or local path (default: slseanwu/MIDI-LLM_Llama-3.2-1B)"
)
# Input arguments (not required if using --interactive only)
@@ -494,4 +495,3 @@ Examples:
if __name__ == "__main__":
main()