zhvng / open-musiclm

Implementation of MusicLM, a text to music model published by Google Research, with a few modifications.
https://arxiv.org/abs/2301.11325
MIT License
511 stars 59 forks source link

Pretrained models #1

Closed Saltb0xApps closed 1 year ago

Saltb0xApps commented 1 year ago

Hi there, I want to start off by saying amazing work replicating MusicLM so quickly!

I'm trying to get this to run on my local device, but it seems like that involves downloading the 100Gb fma_large dataset, followed by training all the various stages. Would it be possible for you to upload the pretrained models in the results folders so everyone can try this out quickly?

zhvng commented 1 year ago

Hey! This repo is a work in progress and I'm still verifying that the implementation works. Will upload checkpoints once I am able to train it and obtain preliminary results.