skytnt commited on
Commit
b37778c
1 Parent(s): 2e60fd4

duration for onnx

Browse files
Files changed (1) hide show
  1. app_onnx.py +2 -2
app_onnx.py CHANGED
@@ -216,9 +216,9 @@ def send_msgs(msgs):
216
  def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
217
  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
218
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
219
- t = gen_events // 23
220
  if "large" in model_name:
221
- t = gen_events // 14
222
  return t + 5
223
 
224
 
 
216
  def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
217
  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
218
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
219
+ t = gen_events // 30
220
  if "large" in model_name:
221
+ t = gen_events // 23
222
  return t + 5
223
 
224