Update app.py

app.py

@@ -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):
+        with torch.amp.autocast(device_type='cuda', dtype=torch.bfloat16):
             out = model.module.generate(inp,
                                         1,
                                         temperature=0.9,