Skip to content

Prompt

Authors

Victor Shepardson Intelligent Instruments Lab 2024

main(midi_file, checkpoint='notochord-latest.ckpt', start_tokens=True)

Run the contents of a MIDI file through a Notochord model and cache the results.

Parameters:

Name Type Description Default
midi_file

path to a MIDI file to prompt with.

 required
checkpoint

path to Notochord model.

 'notochord-latest.ckpt'
start_tokens

whether to include send start-of-sequence tokens in prompt

 True
Source code in src/notochord/app/prompt.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def main(
        midi_file, 
        checkpoint="notochord-latest.ckpt",
        start_tokens=True
        ):
    """
    Run the contents of a MIDI file through a Notochord model and cache the results.

    Args:
        midi_file: path to a MIDI file to prompt with.
        checkpoint: path to Notochord model.
        start_tokens: whether to include send start-of-sequence tokens in prompt
    """
    print(f'{midi_file=} {checkpoint=}')
    noto = Notochord.from_checkpoint(checkpoint)
    noto.eval()
    noto.reset(start=start_tokens)
    _, inst_map = noto.prompt(midi_file)
    print('MIDI channel to instrument:', inst_map)