forcing torch.load to CPU

app.py

@@ -7,7 +7,13 @@ import soundfile as sf
 import torch
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 _old_load = torch.load
-torch.load = lambda *a, **k: _old_load(*a, **({'map_location': device} | k if 'map_location' not in k else k))
+
+def safe_torch_load(*args, **kwargs):
+    if 'map_location' not in kwargs:
+        kwargs['map_location'] = device
+    return _old_load(*args, **kwargs)
+
+torch.load = safe_torch_load
 
 import torchaudio
 import hydra