Why dont we adapt this repo to use cpu in case the user doesnt have a nvidia gpu?

[hmassareli]

some parts of the code are trying to check if there is a cuda compatible gpu before installing gpu, but some other parts are just hardcoding 'cpu' on torch.load...