Guided-Accompaniment-Transformer

Running on Zero

asigalov61 commited on Nov 6, 2024

Commit

8d92046

verified ·

1 Parent(s): 932ddff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -280,7 +280,7 @@ def generate_music(prime,
         drum_pitch = random.choice(drums)
         inputs.extend([0, ((8*8)+6)+256, ((128*129)+drum_pitch)+2304])
-    torch.cuda.empty_cache()
     model.cuda()
     model.eval()

         drum_pitch = random.choice(drums)
         inputs.extend([0, ((8*8)+6)+256, ((128*129)+drum_pitch)+2304])
+    # torch.cuda.empty_cache()
     model.cuda()
     model.eval()