Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ def load_model(model_selector):
|
|
| 102 |
|
| 103 |
model_checkpoint = hf_hub_download(repo_id='asigalov61/Monster-Piano-Transformer', filename=MODEL_CHECKPOINTS[model_selector])
|
| 104 |
|
| 105 |
-
model.load_state_dict(torch.load(model_checkpoint, map_location='cpu'))
|
| 106 |
|
| 107 |
model = torch.compile(model, mode='max-autotune')
|
| 108 |
|
|
|
|
| 102 |
|
| 103 |
model_checkpoint = hf_hub_download(repo_id='asigalov61/Monster-Piano-Transformer', filename=MODEL_CHECKPOINTS[model_selector])
|
| 104 |
|
| 105 |
+
model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))
|
| 106 |
|
| 107 |
model = torch.compile(model, mode='max-autotune')
|
| 108 |
|