add asigalov61/Music-Llama-Medium

app.py

@@ -183,8 +183,8 @@ def run(model_name, tab, mid_seq, instruments, drum_kit, bpm, mid, midi_events,
     yield mid_seq, None, None, seed, send_msgs(init_msgs)
     t = time.time() + 1
     midi_generator = generate(model, mid, max_len=max_len, temp=temp, top_p=top_p, top_k=top_k,
-                         disable_patch_change=disable_patch_change, disable_control_change=not allow_cc,
-                         disable_channels=disable_channels, generator=generator)
+                              disable_patch_change=disable_patch_change, disable_control_change=not allow_cc,
+                              disable_channels=disable_channels, generator=generator)
     events = []
     for i, token_seq in enumerate(midi_generator):
         token_seq = token_seq.tolist()
@@ -192,7 +192,8 @@ def run(model_name, tab, mid_seq, instruments, drum_kit, bpm, mid, midi_events,
         events.append(tokenizer.tokens2event(token_seq))
         ct = time.time()
         if ct - t > 0.5:
-            yield mid_seq, None, None, seed, send_msgs([create_msg("visualizer_append", events), create_msg("progress", [i + 1, gen_events])])
+            yield mid_seq, None, None, seed, send_msgs(
+                [create_msg("visualizer_append", events), create_msg("progress", [i + 1, gen_events])])
             t = ct
             events = []
 
@@ -247,6 +248,7 @@ def hf_hub_download_retry(repo_id, filename):
     if err:
         raise err
 
+
 def get_tokenizer(config_name):
     tv, size = config_name.split("-")
     tv = tv[1:]
@@ -261,6 +263,7 @@ def get_tokenizer(config_name):
     tokenizer.set_optimise_midi(o)
     return tokenizer
 
+
 number2drum_kits = {-1: "None", 0: "Standard", 8: "Room", 16: "Power", 24: "Electric", 25: "TR-808", 32: "Jazz",
                     40: "Blush", 48: "Orchestra"}
 patch2number = {v: k for k, v in MIDI.Number2patch.items()}
@@ -273,10 +276,11 @@ if __name__ == "__main__":
     parser.add_argument("--max-gen", type=int, default=1024, help="max")
     opt = parser.parse_args()
     soundfont_path = hf_hub_download_retry(repo_id="skytnt/midi-model", filename="soundfont.sf2")
-    models_info = {"generic pretrain model (tv2o-large)": ["asigalov61/Music-Llama", "", "tv2o-large"],
-                   "generic pretrain model (tv1-medium)": ["skytnt/midi-model", "", "tv1-medium"],
-                   "j-pop finetune model (tv1-medium)": ["skytnt/midi-model-ft", "jpop/", "tv1-medium"],
-                   "touhou finetune model (tv1-medium)": ["skytnt/midi-model-ft", "touhou/", "tv1-medium"],
+    models_info = {"generic pretrain model (tv2o-large) by asigalov61": ["asigalov61/Music-Llama", "", "tv2o-large"],
+                   "generic pretrain model (tv2o-medium) by asigalov61": ["asigalov61/Music-Llama-Medium", "", "tv2o-medium"],
+                   "generic pretrain model (tv1-medium) by skytnt": ["skytnt/midi-model", "", "tv1-medium"],
+                   "j-pop finetune model (tv1-medium) by skytnt": ["skytnt/midi-model-ft", "jpop/", "tv1-medium"],
+                   "touhou finetune model (tv1-medium) by skytnt": ["skytnt/midi-model-ft", "touhou/", "tv1-medium"],
                    }
     models = {}
     providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
@@ -318,8 +322,8 @@ if __name__ == "__main__":
                 input_drum_kit = gr.Dropdown(label="🥁drum kit", choices=list(drum_kits2number.keys()), type="value",
                                              value="None")
                 input_bpm = gr.Slider(label="BPM (beats per minute, auto if 0)", minimum=0, maximum=255,
-                                              step=1,
-                                              value=0)
+                                      step=1,
+                                      value=0)
                 example1 = gr.Examples([
                     [[], "None"],
                     [["Acoustic Grand"], "None"],
@@ -340,8 +344,10 @@ if __name__ == "__main__":
                                               step=1,
                                               value=128)
                 input_reduce_cc_st = gr.Checkbox(label="reduce control_change and set_tempo events", value=True)
-                input_remap_track_channel = gr.Checkbox(label="remap tracks and channels so each track has only one channel and in order", value=True)
-                input_add_default_instr = gr.Checkbox(label="add a default instrument to channels that don't have an instrument", value=True)
+                input_remap_track_channel = gr.Checkbox(
+                    label="remap tracks and channels so each track has only one channel and in order", value=True)
+                input_add_default_instr = gr.Checkbox(
+                    label="add a default instrument to channels that don't have an instrument", value=True)
                 input_remove_empty_channels = gr.Checkbox(label="remove channels without notes", value=False)
                 example2 = gr.Examples([[file, 128] for file in glob.glob("example/*.mid")],
                                        [input_midi, input_midi_events])
@@ -375,7 +381,7 @@ if __name__ == "__main__":
                                         input_top_k, input_allow_cc],
                                   [output_midi_seq, output_midi, output_audio, input_seed, js_msg],
                                   concurrency_limit=3)
-        stop_btn.click(cancel_run, [input_model,output_midi_seq],
+        stop_btn.click(cancel_run, [input_model, output_midi_seq],
                        [output_midi, output_audio, js_msg],
                        cancels=run_event, queue=False)
-    app.launch(server_port=opt.port, share=opt.share, inbrowser=True)
+    app.launch(server_port=opt.port, share=opt.share, inbrowser=True)