update tokenizer

app.py

@@ -311,7 +311,7 @@ if __name__ == "__main__":
  step=1,
  value=128)
  input_reduce_cc_st = gr.Checkbox(label="reduce control_change and set_tempo events", value=True)
- input_remap_track_channel = gr.Checkbox(label="remap tracks and channels to have only one channel per track", value=True)
+ input_remap_track_channel = gr.Checkbox(label="remap tracks and channels so each track has only one channel and in order", value=True)
  input_add_default_instr = gr.Checkbox(label="add a default instrument to channels that don't have an instrument", value=True)
  input_remove_empty_channels = gr.Checkbox(label="remove channels without notes", value=False)
  example2 = gr.Examples([[file, 128] for file in glob.glob("example/*.mid")],

midi_tokenizer.py

@@ -131,6 +131,8 @@ class MIDITokenizer:
  patch_channels = []
  channels_count = 0
  channels_map = {9: 9} if 9 in channels else {}
+ if remove_empty_channels:
+ channels = sorted(channels, key=lambda x: 1 if x in empty_channels else 0)
  for c in channels:
  if c == 9:
  continue