Update app.py

app.py

@@ -71,8 +71,8 @@ dtype = 'bfloat16'
 ptdtype = {'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
 ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype)
 
-SEQ_LEN = 1536
-PAD_IDX = 708
+SEQ_LEN = 8192
+PAD_IDX = 18819
 
 model = TransformerWrapper(num_tokens = PAD_IDX+1,
                            max_seq_len = SEQ_LEN,
@@ -210,7 +210,7 @@ def Generate_Chords_Textures(input_midi,
     else:
         seq = [705] + chords[:128] + [706]
 
-    x = torch.LongTensor(seq).cuda()
+    x = torch.LongTensor(seq).to(device_type)
     
     with ctx:
         out = model.generate(x,