Music-text contrastive (CLAP) models trained with captions derived from professional album reviews, plus the AllMusicCaps caption dataset.

Install

From PyPI

pip install amclap

From source

For embedding extraction or fine-tuning:

pip install .

For development including pre-training your own models:

pip install -e .[train]

Inference

Load a model by specifying its Hugging Face model ID:

import torch
from amclap import get_model

x_a = torch.randn(1, 24000 * 10).cpu()   # mono audio at 24 kHz
x_t = ["dreamy shoegaze with washed-out guitars"]

model_id = "mtg-upf/allmusiccaps_te_trained_sigreg"
model = get_model(model_id=model_id, device="cpu")

with torch.no_grad():
    z_a = model.forward_audio(x_a)       # torch.Size([1, 512])
    z_t = model.forward_text(x_t)        # torch.Size([1, 512])

Audio must be mono at 24 kHz. Both towers output 512-dimensional embeddings in a shared space, comparable with cosine similarity.

Note: pip install amclap is enough for inference on every released model. Training with the SigReg objective additionally needs lejepa, which has no PyPI release and so is not part of the [train] extra — install it from git:

pip install "lejepa @ git+https://github.com/rbalestr-lab/lejepa.git"

Available models

All models use the OMAR-RQ audio encoder and an all-mpnet-base-v2 text encoder (TE). Step is the checkpoint the paper reports.

te_trained_sigreg is the best overall model; all_layers is the frozen-TE recipe it builds on.

Note: models with a trainable text encoder overfit past ~40--80k steps, so they are released at their 60k checkpoint rather than the final one.

Hugging Face Model IDs

• mtg-upf/allmusiccaps_baseline
• mtg-upf/allmusiccaps_amcquotes
• mtg-upf/allmusiccaps_amcstruct
• mtg-upf/allmusiccaps_baseline_quotes
• mtg-upf/allmusiccaps_baseline_struct
• mtg-upf/allmusiccaps_layer6
• mtg-upf/allmusiccaps_all_layers
• mtg-upf/allmusiccaps_sigmoid
• mtg-upf/allmusiccaps_lejepa
• mtg-upf/allmusiccaps_infonce_sigreg
• mtg-upf/allmusiccaps_te_trained
• mtg-upf/allmusiccaps_te_trained_sigreg

The AllMusicCaps dataset

540,454 rows pairing YouTube tracks with captions derived from AllMusic album reviews, in two styles: review quotes and LLM-filled structured attributes. It contains identifiers and captions, no audio.

from datasets import load_dataset

ds = load_dataset("mtg-upf/allmusiccaps", split="train")
print(ds[0]["generated_quotes_captions"])

v1 is the default. allmusiccaps_v0.jsonl is what the paper's models were trained on, kept for exact reproducibility: it holds the raw LLM output, in which generated_quotes_captions is not consistently typed, so Arrow-backed readers reject it. v1 normalizes that field to list[string] and changes nothing else. See the dataset card for the mapping and how to read v0, and normalize_allmusiccaps_v1.py to reproduce it. A 50-row sample lives in data/allmusiccaps/_samples/.

Training

1. Install development dependencies:

pip install -e .[train]

1. Prepare the data

Audio is stored downsampled to 24 kHz mono as 16-bit raw bytes (numpy memmap files); captions come from the JSONL above. Check the preprocessing scripts.

1. Configuration

Experiment configuration is controlled with gin-config; see cfg/README.md. At least the dataset paths need to point at your own data.

1. Run the experiment

python -m amclap.train cfg/<config>.gin

Citation

If you find this work useful, please cite the paper:

@inproceedings{alonso2026allmusiccaps,
  title = {{AllMusicCaps}: Album Reviews as Complementary Supervision for Music {CLAP}},
  author = {Alonso-Jim{\'e}nez, Pablo and Lizarraga-Seijas, Xavier and Serra, Xavier and Bogdanov, Dmitry},
  booktitle = {International Society for Music Information Retrieval Conference (ISMIR)},
  year = {2026},
}

Licensing information

The code in this repository is available under AGPL-3.0 license. The model weights are available under CC BY-NC-SA 4.0 license for non-commercial applications. The AllMusicCaps dataset is released for non-commercial scientific research purposes only, and any publication of results based on it must cite AllMusic as the source of the data. Contact us for more information.