Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on Apr 4

Commit

16448c2

•

1 Parent(s): 399d36d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -127,7 +127,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
         inp = torch.LongTensor([outy]).cuda()
-        with torch.amp.autocast(device_type='cuda', dtype=torch.float16)
             out = model.module.generate(inp,
                                         1,
                                         temperature=0.9,

         inp = torch.LongTensor([outy]).cuda()
+        with torch.amp.autocast(device_type='cuda', dtype=torch.float16):
             out = model.module.generate(inp,
                                         1,
                                         temperature=0.9,