Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on 23 days ago

Commit

5aa99eb

•

1 Parent(s): 889f588

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -122,7 +122,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.bfloat16):
             out = model.module.generate(inp,
                                         1,
                                         temperature=0.9,