forcing torch.load to CPU

app.py

@@ -5,9 +5,9 @@ import soundfile as sf
 
 # forcing torch.load to CPU
 import torch
-_old_load = torch.load
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-torch.load = lambda *a, **k: _old_load(*a, map_location=k.pop('map_location', device))
+_old_load = torch.load
+torch.load = lambda *a, **k: _old_load(*a, **({'map_location': device} | k if 'map_location' not in k else k))
 
 import torchaudio
 import hydra