Upload scripts/end_to_end/minimax_ttm_test.py with huggingface_hub

scripts/end_to_end/minimax_ttm_test.py

@@ -0,0 +1,173 @@
+# SPDX-License-Identifier: Apache-2.0
+"""Generate a MiniMax Music 3 WAV through the SGLang-Omni HTTP API."""
+
+from __future__ import annotations
+
+import argparse
+import json
+import urllib.error
+import urllib.request
+from pathlib import Path
+
+CAPTION = """Global Metadata
+Basic Attributes: bpm is 92. key is E, and scale is minor. Electric Blues / Blues Rock.
+Global Emotional Progression: The track establishes a confident, gritty swagger from the outset, maintaining a driving, mid-tempo groove throughout. The emotional intensity fluctuates dynamically between the restrained, storytelling verses and the explosive, high-energy instrumental breaks where the lead guitar takes center stage. The song builds tension through call-and-response interactions before releasing it in extended, virtuosic solos, culminating in a frenetic and improvisational outro that fades on a sustained note of raw energy.
+Application Scenarios & Imagery: A smoky, dimly lit blues club late at night, a road trip down a dusty highway in a classic muscle car, or a scene depicting a charismatic character making a bold entrance in a film.
+Sonics & Production Profile: The production favors a live, organic feel with a moderately wide soundstage that places the lead guitar slightly off-center for prominence. The frequency response is warm and mid-range heavy, emphasizing the grit of the guitar tubes and the body of the vocals, while the low end remains tight but not overpowering. The dynamic aesthetic is relatively uncompressed, preserving the natural attack of the drums and the nuances of the guitar playing, giving the mix a "roomy" and authentic atmosphere.
+Vocal Details
+Vocal Gender & Timbre: Singer A (Male). The vocalist possesses a deep, gravelly baritone with a raspy, textured quality that conveys age, experience, and authority.
+Vocal Style: The delivery is conversational and storytelling-oriented, often bordering on spoken-word phrasing in the verses to emphasize the narrative. This shifts into a more melodic, soulful singing style during the refrains, punctuated by energetic ad-libs, shouts, and exclamations that heighten the excitement during instrumental peaks.
+Harmony/Backing Vocals: There are no distinct backing vocalists; however, the lead singer occasionally layers his own voice or utilizes call-and-response phrasing with his guitar to create a dialogue effect, particularly during the bridge and solo sections.
+Vocal FX: The vocal track is treated with a moderate amount of plate reverb to add depth and space, along with a subtle slapback delay that enhances the vintage blues aesthetic without obscuring the lyrical clarity.
+Arrangement
+Instrument Lifecycle Description (Primary/Secondary Layering):
+Primary: A clean-to-slightly-overdriven electric guitar serves as the primary melodic and harmonic driver, present from the Intro to the Outro. It alternates between rhythmic chordal comping and searing, expressive lead lines that dominate the instrumental sections.
+Secondary: A walking electric bass line provides a solid, melodic foundation throughout the entire track, locking tightly with the drums. A Hammond-style organ enters subtly in the background during the verses, swelling in volume during the choruses and solos to add harmonic thickness and warmth.
+Groove & Foundation Progression: A standard acoustic drum kit drives the rhythm with a steady shuffle beat, characterized by a crisp snare on beats 2 and 4 and a swinging hi-hat pattern. The groove remains consistent in tempo but increases in perceived intensity during the solos as the drummer adds more fills and dynamic accents on the crash cymbals.
+Embellishments, Textures & Spatial FX: Occasional guitar slides and string squeals act as transitional flourishes between lyrical phrases. The arrangement features significant use of "space," allowing the instruments to breathe during the verses, which contrasts with the dense, layered texture of the extended guitar solos where the organ and rhythm section push forward to support the lead."""
+
+LYRICS = """[verse]
+I’m learning how to fill up
+every space I used to leave,
+teaching my own heart the patience
+that my family needs from me.
+And I’m learning how to soften,
+how to rise above the past—
+Ever’s gonna need a father
+whose love is built to last.
+[pre-chorus]
+Breathe a little deeper,
+love is here to heal.
+Let love lift what used to fall—
+breathe a little deeper, yeah.
+Love is here to heal,
+love is here to heal,
+let love lift what used to fall.
+[verse]
+When I heard your heartbeat for the very first time,
+it broke me open deep inside.
+Pinched my eyes between my fingers,
+’cause that’s when I realized—
+you were healing what was numb in me,
+every beat just whisperin’ “overcome.”
+[pre-chorus]
+Breathe a little deeper,
+love is here to heal.
+Let love lift what used to fall—
+breathe a little deeper, yeah.
+[bass-quartet-rumbles-in]
+Every heartbeat sounded like,
+“We shall overcome, Daddy…
+let’s heal what once was numb.”
+And the joy outran the fear,
+washed away the hurt and pain—
+[pre-chorus]
+Breathe a little deeper,
+love is here to heal.
+Breathe a little deeper,
+love is here to heal.
+Love is here to heal,
+let love lift what used to fall…
+Breathe a little deeper,
+love is here to heal.
+Love is here to heal,
+love is here to heal.
+[bridge]
+Every time I see you wiggling there,
+I’m reminded again and again.
+What keeps playing on repeat
+in my spirit, in my mind,
+is the sound of your heartbeat…
+and I can’t wait to hold you next to mine.
+[interlude]
+You gotta let love— (Oh love… lift us up…)
+Love Lift us up higher—-
+(oh my sweet little Princess Ever Don’t EVER ever ever fear the things you might not yet understand about yourself or father and mother or this world)—
+As long as you let love— Oh love—
+Continue DOing all your heavy lifting Ever-Michelle
+nothing else matters
+when you let love— oh love—
+lift whatever fell or falls…
+[chorus]
+You gotta let love— oh love—Let love—-oh love—-
+Lift what used to fall…
+You gotta let love— oh love—Let love—-oh love—-
+Lift what used to fall…
+You gotta let love— oh love—Let love—-oh love—-
+Lift what used to fall…
+[outro]
+We’re gonna let love stay,
+let our vows endure.
+Oh love… stay… stay…
+We’re gonna let love stay,
+’cause this home is built with love.
+Oh love… stay… stay…
+We’re gonna let love stay,
+let our vows endure—
+this home is made with love,
+and love is here to stay."""
+
+
+DEFAULT_OUT = Path("minimax_ttm.wav")
+
+
+def build_request_body(*, seed: int, max_frames: int) -> dict[str, object]:
+    return {
+        "model": "MiniMaxAI/MiniMax-Music3",
+        "input": LYRICS,
+        "instructions": CAPTION,
+        "response_format": "wav",
+        "seed": seed,
+        "max_new_tokens": max_frames,
+        "stream": False,
+    }
+
+
+def _parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("--server-url", default="http://127.0.0.1:8000")
+    parser.add_argument("--out", type=Path, default=DEFAULT_OUT)
+    parser.add_argument("--seed", type=int, default=0)
+    parser.add_argument("--max-frames", type=int, default=9000)
+    parser.add_argument("--timeout", type=float, default=1200)
+    return parser
+
+
+def _post(url: str, body: dict[str, object], timeout: float) -> bytes:
+    request = urllib.request.Request(
+        url,
+        data=json.dumps(body, ensure_ascii=False).encode("utf-8"),
+        headers={"Content-Type": "application/json"},
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(request, timeout=timeout) as response:
+            return response.read()
+    except urllib.error.HTTPError as exc:
+        raise RuntimeError(
+            f"HTTP {exc.code}: {exc.read().decode(errors='replace')}"
+        ) from exc
+    except urllib.error.URLError as exc:
+        raise RuntimeError(f"cannot connect to {url}: {exc}") from exc
+
+
+def main() -> int:
+    args = _parser().parse_args()
+    if args.max_frames < 1:
+        raise SystemExit("--max-frames must be positive")
+    wav = _post(
+        args.server_url.rstrip("/") + "/v1/audio/speech",
+        build_request_body(seed=args.seed, max_frames=args.max_frames),
+        args.timeout,
+    )
+    if not wav:
+        raise RuntimeError("server returned an empty response")
+    output = args.out.expanduser().resolve()
+    output.parent.mkdir(parents=True, exist_ok=True)
+    output.write_bytes(wav)
+    print(f"WAV written to: {output}")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())