Orpheus-Humanizing-Transformer

Running on Zero

projectlosangeles commited on Jun 20

Commit

958c5a5

verified ·

1 Parent(s): 50c36d3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,7 +56,7 @@ print('=' * 70)
 #==================================================================================
-MODEL_CHECKPOINT = 'Orpheus_Bridge_Music_Transformer_Trained_Model_19571_steps_0.9396_loss_0.7365_acc.pth'
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -71,7 +71,7 @@ dtype = 'bfloat16'
 ptdtype = {'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
 ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype)
-SEQ_LEN = 1668
 PAD_IDX = 18819
 model = TransformerWrapper(num_tokens = PAD_IDX+1,

 #==================================================================================
+MODEL_CHECKPOINT = 'Orpheus_Music_Transformer_Trained_Model_60667_steps_0.2238_loss_0.9375_acc.pth'
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 ptdtype = {'bfloat16': torch.bfloat16, 'float16': torch.float16}[dtype]
 ctx = torch.amp.autocast(device_type=device_type, dtype=ptdtype)
+SEQ_LEN = 8192
 PAD_IDX = 18819
 model = TransformerWrapper(num_tokens = PAD_IDX+1,