map_location argument is not supported

[Mikkolehtimaki]

Typically it's possible to load torch models to cpu / gpu by using the map_location argument.

This doesn't work for the acoustic model:

TypeError: hubert_soft() got an unexpected keyword argument 'map_location

On a CPU-only machine loading this model gives the error:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu')
 to map your storages to the CPU.