Run SAM on CPU

[Minh Tran]

Hi, what are the changes I need to make to run this on CPU? Thank you very much!
I've tried use 'cpu' instead of 'cuda', but I got this error while importing pSp:
ImportError: /root/.cache/torch_extensions/py37_cu113/fused/fused.so: cannot open shared object file: No such file or directory

[yuval-alaluf]

You will need to replace the usage of the CUDA kernels that are used by default with a native implementation.
I hope the following thread can help: rosinality/stylegan2-pytorch#89

[Minh Tran]

Thank you, I got it up and run, making a pull request soon!

[Hayk]

@minhtcai could you share your version that runs on CPU?