inference on apple silicon M2

[ankan2709]

I am using macbook M2 and when i run the code on the jupyter notebook to load the model, I am getting this 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.

i tried this avatar = NHAOptimizer.load_from_checkpoint(ckpt, map_location=torch.device("cpu")).eval()
but it still does not work.