metadata
library_name: transformers
tags: []
Model based on facebook/opt-125m trained on small dataset to generate music from old games in midi format.
Usage
!pip install miditok==3.0.2
from transformers import AutoModel, AutoModelForCausalLM
from miditok import MIDITokenizer
tokenizer = MIDITokenizer.from_pretrained('lklimkiewicz/midi-ganerator-game')
model = AutoModel.from_pretrained('lklimkiewicz/midi-ganerator-game', trust_remote_code=True).cuda()
music = model.generate_music(tokenizer, temperature=0.1)
music.dump_midi('out.mid')
!timidity out.mid