changes

app.py

@@ -1,5 +1,3 @@
-from concurrent.futures import ThreadPoolExecutor
-
 import spaces
 import random
 import argparse
@@ -7,6 +5,7 @@ import glob
 import json
 import os
 import time
+from concurrent.futures import ThreadPoolExecutor
 
 import gradio as gr
 import numpy as np
@@ -122,7 +121,7 @@ def send_msgs(msgs):
 def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
- t = 1e-4*gen_events**2 + 15
+ t = 1e-4*gen_events**2 + 25
  if "large" in model_name:
  t *= 2
  return t
@@ -383,7 +382,7 @@ if __name__ == "__main__":
  with app:
  gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Midi Composer</h1>")
  gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.midi-composer&style=flat)\n\n"
- "Midi event transformer for music generation\n\n"
+ "Midi event transformer for symbolic music generation\n\n"
  "Demo for [SkyTNT/midi-model](https://github.com/SkyTNT/midi-model)\n\n"
  "[Open In Colab]"
  "(https://colab.research.google.com/github/SkyTNT/midi-model/blob/main/demo.ipynb)"

midi_model.py

@@ -125,7 +125,7 @@ class MIDIModel(nn.Module):
  prompt = np.repeat(prompt, repeats=batch_size, axis=0)
  elif prompt.shape[0] == 1:
  prompt = np.repeat(prompt, repeats=batch_size, axis=0)
- else:
+ elif len(prompt.shape) != 3 or prompt.shape[0] != batch_size:
  raise ValueError(f"invalid shape for prompt, {prompt.shape}")
  prompt = prompt[..., :max_token_seq]
  if prompt.shape[-1] < max_token_seq: