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/MusicFMInferece", trust_remote_code=True)
6>>> # get embeddings
7>>> model.eval()
8>>> emb = model(wav)
9>>> emb.size()
10torch.Size([4, 750, 1024])