skytnt commited on
Commit
469ef25
1 Parent(s): bb51a02
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,9 +100,9 @@ def get_duration(model_name, tab, mid_seq, continuation_state, instruments, drum
100
  key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
101
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
102
  if "large" in model_name:
103
- return gen_events // 10
104
  else:
105
- return gen_events // 20
106
 
107
 
108
  @spaces.GPU(duration=get_duration)
 
100
  key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
101
  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
102
  if "large" in model_name:
103
+ return gen_events // 10 + 10
104
  else:
105
+ return gen_events // 20 + 10
106
 
107
 
108
  @spaces.GPU(duration=get_duration)