asigalov61 commited on
Commit
3b8544d
·
verified ·
1 Parent(s): 8643c09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -357,7 +357,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
357
  filename='Giant_Music_Transformer_Medium_Trained_Model_42174_steps_0.5211_loss_0.8542_acc.pth'
358
  )
359
 
360
- model.load_state_dict(torch.load(kar_model_checkpoint, map_location='cpu', weights_only=True))
361
 
362
  model = torch.compile(model, mode='max-autotune')
363
 
 
357
  filename='Giant_Music_Transformer_Medium_Trained_Model_42174_steps_0.5211_loss_0.8542_acc.pth'
358
  )
359
 
360
+ model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))
361
 
362
  model = torch.compile(model, mode='max-autotune')
363