add optimise midi

app.py

@@ -121,7 +121,7 @@ def send_msgs(msgs):
  return json.dumps(msgs)
 
 
-def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, seed, seed_rand,
+def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, midi_opt, seed, seed_rand,
  gen_events, temp, top_p, top_k, allow_cc):
  mid_seq = []
  bpm = int(bpm)
@@ -153,7 +153,8 @@ def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, seed, see
  disable_patch_change = True
  disable_channels = [i for i in range(16) if i not in patches]
  elif mid is not None:
- mid = tokenizer.tokenize(MIDI.midi2score(mid))
+ eps = 4 if midi_opt else 0
+ mid = tokenizer.tokenize(MIDI.midi2score(mid), cc_eps=eps, tempo_eps=eps)
  mid = np.asarray(mid, dtype=np.int64)
  mid = mid[:int(midi_events)]
  for token_seq in mid:
@@ -305,6 +306,7 @@ if __name__ == "__main__":
  input_midi_events = gr.Slider(label="use first n midi events as prompt", minimum=1, maximum=512,
  step=1,
  value=128)
+ input_midi_opt = gr.Checkbox(label="optimise midi", value=True)
  example2 = gr.Examples([[file, 128] for file in glob.glob("example/*.mid")],
  [input_midi, input_midi_events])
 
@@ -328,8 +330,8 @@ if __name__ == "__main__":
  output_audio = gr.Audio(label="output audio", format="mp3", elem_id="midi_audio")
  output_midi = gr.File(label="output midi", file_types=[".mid"])
  run_event = run_btn.click(run, [input_model, tab_select, input_instruments, input_drum_kit, input_bpm,
- input_midi, input_midi_events, input_seed, input_seed_rand, input_gen_events,
- input_temp, input_top_p, input_top_k, input_allow_cc],
+ input_midi, input_midi_events, input_midi_opt, input_seed, input_seed_rand,
+ input_gen_events, input_temp, input_top_p, input_top_k, input_allow_cc],
  [output_midi_seq, output_midi, output_audio, input_seed, js_msg],
  concurrency_limit=3)
  stop_btn.click(cancel_run, [output_midi_seq], [output_midi, output_audio, js_msg], cancels=run_event, queue=False)

requirements.txt

@@ -1,5 +1,5 @@
 Pillow
 numpy
 onnxruntime-gpu
-gradio==4.36.1
+gradio==4.43.0
 pyfluidsynth