| --- |
| license: mit |
| library_name: pytorch |
| tags: |
| - music-generation |
| - symbolic-music |
| - abc-notation |
| - classical |
| - transformer |
| pipeline_tag: text-to-audio |
| --- |
| |
| # NotaGen-X — MAESTRO Mirror |
|
|
| Safetensors mirror of the NotaGen-X model weights, used by |
| [MAESTRO](https://github.com/AEmotionStudio) for the AI-Workstation "Create → |
| NotaGen" model card. |
|
|
| This repository ships **only the model weights** (no optimizer state) in the |
| [safetensors](https://github.com/huggingface/safetensors) format, so they load |
| into MAESTRO without arbitrary-code execution risk. The original upstream |
| checkpoints from `ElectricAlexis/NotaGen` are PyTorch `.pth` files bundling |
| optimizer state, epoch counters, etc. |
|
|
| ## Files |
|
|
| | File | Purpose | |
| | --- | --- | |
| | `model.safetensors` | NotaGen-X model state dict (516 M params, fp32, ~1.92 GB) | |
| | `config.json` | Architecture hyperparameters consumed by NotaGen's loader | |
| | `README.md` | This file | |
|
|
| ## Provenance |
|
|
| * Upstream: |
| [`ElectricAlexis/NotaGen`](https://huggingface.co/ElectricAlexis/NotaGen) |
| · file `weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth` |
| * Conversion: `torch.load(...)['model']` → contiguous CPU tensors → |
| `safetensors.torch.save_file(...)`. Round-trip verified (zero mismatches over |
| all 323 tensors). |
| * Training metadata preserved in safetensors header |
| (`epoch`, `best_epoch`, `min_eval_loss`). |
|
|
| ## Architecture |
|
|
| Hierarchical Tunesformer (GPT-2 backbone) generating ABC notation: |
|
|
| * `p_size = 16` (patch size) |
| * `p_length = 1024` (patch sequence length) |
| * `p_layers = 20` (patch-level decoder layers) |
| * `c_layers = 6` (character-level decoder layers) |
| * `h_size = 1280` (hidden size) |
|
|
| ## Loading |
|
|
| ```python |
| from safetensors.torch import load_file |
| state = load_file("model.safetensors") |
| # Pass to NotaGen's model class — same key layout as the upstream `.pth`'s |
| # `["model"]` sub-dict. |
| ``` |
|
|
| MAESTRO's loader (`backend/ai/models/notagen.py`) handles this automatically and, |
| if the upstream NotaGen inference script demands a `.pth` on disk, materialises |
| one alongside the safetensors on first load. |
|
|
| ## License |
|
|
| MIT, inherited from upstream NotaGen. |
|
|
