Run the model in CPU

[akote123]

I want to run the unet model in cpu. I have tried
net = torch.hub.load('milesial/Pytorch-UNet', 'unet_carvana', pretrained=True, scale=0.5)
But I am getting error as map this to cpu when I try to add mappings in torch.hub.load argument it says the argument is not there.
So how can I run this model in CPU for inference