Downloads · 30 days
0
brianflakes/rwkv-midi-piano
rwkv-midi-piano is a machine learning model from brianflakes. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
step650 model was trained in 48hr on an RTX3090, step1100 model initially experimentally trained to add tiny attenion (RWKV-4b), but then finetuned again to remove it. Final finetuning was done with ctx 4096.
Downloads · 30 days
0
Access
Public
Updated May 29, 2023
Repo size
1.3 GB
Likes
18
Public
Click a slice to open those files.
.pth1.3 GB · 100%
From the Hugging Face model README
step650 model was trained in 48hr on an RTX3090, step1100 model initially experimentally trained to add tiny attenion (RWKV-4b), but then finetuned again to remove it. Final finetuning was done with ctx 4096.
Trained using GiantMIDI-Piano dataset, augmented 30x. See training history: https://wandb.ai/brianflakes/rwkv-midi
To run, use the rwkv pip package: https://pypi.org/project/rwkv/
To convert outputs to MIDI, use https://github.com/briansemrau/MIDI-LLM-tokenizer