skytnt commited on
Commit
4789653
1 Parent(s): 0db6e6a
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -128,10 +128,10 @@ def send_msgs(msgs):
128
  def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
129
  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
130
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
131
- t = gen_events // 20 + 5
132
  if "large" in model_name:
133
- t *= 2
134
- return t
135
 
136
 
137
  @spaces.GPU(duration=get_duration)
 
128
  def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
129
  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
130
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
131
+ t = gen_events // 23
132
  if "large" in model_name:
133
+ t = gen_events // 14
134
+ return t + 5
135
 
136
 
137
  @spaces.GPU(duration=get_duration)