Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|
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 |
|