---
license: apache-2.0
datasets:
- projectlosangeles/Los-Angeles-MIDI-Dataset
library_name: transformers
tags:
- music
- midi
- music generation
- midi generation
---
# Model Card for midi-model

Midi event transformer for music generation

### Model Updates

- tv2o-medium model: MidiTokenizerV2. [skytnt/midi-model-tv2o-medium](https://huggingface.co/skytnt/midi-model-tv2o-medium)
- v1.2 : Optimise the tokenizer and dataset. The dataset was filtered by MIDITokenizer.check_quality. Using the higher quality dataset to train the model, the performance of the model is significantly improved.

## Model Details

### Model Description

- **Developed by:** SkyTNT
- **Model type:** Transformer
- **License:** apache-2.0

### Model Sources

- **Repository:** https://github.com/SkyTNT/midi-model
- **Demo:** https://huggingface.co/spaces/skytnt/midi-composer


## Training Details

### Training Data

- [projectlosangeles/Los-Angeles-MIDI-Dataset](https://huggingface.co/datasets/projectlosangeles/Los-Angeles-MIDI-Dataset)

#### Training Hyperparameters

- basemodel: v1.1
- lr: 1e-4
- weight-decay: 0.01
- batch: 2x2x2
- fp16 mixed precision

####  Loss
 - val loss: 0.238

## Files

- model.ckpt: latest model checkpoint
- soundfont.sf2: soundfont file, i put it here for convenient downloading , [source](https://github.com/musescore/MuseScore/tree/master/share/sound)
- onnx/*.onnx: onnx format model