Views
No views yet
transformers.AutoModel.1>>> import torch
2>>> from transformers import AutoModel
3>>> torch.manual_seed(0)
4>>> wav = (torch.rand(4, 24000 * 30) - 0.5) * 2
5>>> model = AutoModel.from_pretrained("tky823/MusicFM", trust_remote_code=True)
6>>> # get embeddings
7>>> model.eval()
8>>> emb = model.get_latent(wav, layer_ix=7)
9>>> emb.size()
10torch.Size([4, 750, 1024])