You can deploy this model and start earning money today!

magenta-community/
magenta-rt-musiccoca-torch

Name: magenta-rt-musiccoca-torch
Availability: InStock

Views

No views yet

MusicCoCa (text path) — PyTorch / TorchScript

Pure-torch conversion of the MusicCoCa text style encoder from google/magenta-realtime-2 (upstream ships it only as TFLite). Converted TFLite → ONNX (tf2onnx) → torch (onnx2torch + the custom op shim in convert_ops.py) and traced to TorchScript.

text_encoder.pt — text tower: (ids[1,128] int32, paddings[1,128] f32) → [1,768]
quantizer.pt — RVQ: [1,768] → [12,1] style tokens
spm.model — SentencePiece vocab

Token-exact vs the TFLite reference (validated on 7 prompts → 84/84 tokens). Runtime deps: torch + sentencepiece only. Use via magenta_rt.torch.musiccoca.MusicCoCa.

magenta-community/magenta-rt-musiccoca-torch

MusicCoCa (text path) — PyTorch / TorchScript

magenta-community/
magenta-rt-musiccoca-torch