Upload inference_trained_model.py

inference_trained_model.py

@@ -203,22 +203,22 @@ def generate(model, tokenizer, midi_tokenizer, metadata, prompt, max_midi_tokens
 def save_midi(midi_score, output_path: str):
     """Save a MidiTok decoded score to a MIDI file.
     
-    MidiTok v3 returns a symusic Score object which uses .write_midi()
+    MidiTok v3 returns a symusic ScoreTick object which uses dump_midi()
     instead of the old miditoolkit MidiFile.dump() method.
     """
     out = Path(output_path)
     out.parent.mkdir(parents=True, exist_ok=True)
     
-    # MidiTok v3 (symusic backend) uses write_midi()
-    if hasattr(midi_score, "write_midi"):
-        midi_score.write_midi(str(out))
+    # symusic ScoreTick (MidiTok v3) uses dump_midi()
+    if hasattr(midi_score, "dump_midi"):
+        midi_score.dump_midi(str(out))
     # Fallback for older miditoolkit MidiFile objects
     elif hasattr(midi_score, "dump"):
         midi_score.dump(str(out))
     else:
         raise AttributeError(
             f"Cannot save MIDI object of type {type(midi_score)}. "
-            "Expected ScoreTick with .write_midi() or MidiFile with .dump()"
+            f"Available attrs: {[a for a in dir(midi_score) if not a.startswith('_')]}"
         )
     
     size = out.stat().st_size