batch 8

app.py

@@ -18,7 +18,7 @@ from midi_model import MIDIModel, MIDIModelConfig
 from midi_synthesizer import MidiSynthesizer
 
 MAX_SEED = np.iinfo(np.int32).max
-OUTPUT_BATCH_SIZE = 4
+OUTPUT_BATCH_SIZE = 8
 in_space = os.getenv("SYSTEM") == "spaces"
 
 
@@ -122,9 +122,9 @@ def get_duration(model_name, tab, mid_seq, continuation_state, continuation_sele
                  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):
     if "large" in model_name:
-        return gen_events // 10 + 15
+        return gen_events // 7 + 5
     else:
-        return gen_events // 20 + 15
+        return gen_events // 15 + 5
 
 
 @spaces.GPU(duration=get_duration)

javascript/app.js

@@ -1,4 +1,4 @@
-const MIDI_OUTPUT_BATCH_SIZE=4;
+const MIDI_OUTPUT_BATCH_SIZE= 8;
 
 /**
  * 自动绕过 shadowRoot 的 querySelector