skytnt commited on
Commit
8f50152
1 Parent(s): 9d28b5c

adjust duration

Browse files
Files changed (1) hide show
  1. app_onnx.py +3 -3
app_onnx.py CHANGED
@@ -216,10 +216,10 @@ 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 // 30
220
  if "large" in model_name:
221
- t = gen_events // 23
222
- return t + 5
223
 
224
 
225
  @spaces.GPU(duration=get_duration)
 
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 // 28
220
  if "large" in model_name:
221
+ t = gen_events // 20
222
+ return t + 10
223
 
224
 
225
  @spaces.GPU(duration=get_duration)