Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on Sep 15, 2023

Commit

1c47239

1 Parent(s): 2a68ddd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -83,7 +83,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     pitch = 0
     channel = 0
-    for i in range(num_tok):
         inp = torch.LongTensor([outy]).cpu()
         out = model.module.generate(inp,

     pitch = 0
     channel = 0
+    for i in range(max(1, min(512, num_tok))):
         inp = torch.LongTensor([outy]).cpu()
         out = model.module.generate(inp,