Edit model card

Model Card for Model ID

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)
music.dump_midi('out.mid')
!timidity out.mid
Downloads last month
2
Safetensors
Model size
110M params
Tensor type
F32
·
Inference Examples
Inference API (serverless) does not yet support model repos that contain custom code.