How can I run this on CPU ?

yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754

https://yuval-alaluf.github.io/SAM/

MIT License

632 stars 151 forks source link

How can I run this on CPU ? #61

Closed kasarampavan closed 1 year ago

kasarampavan commented 1 year ago

when I run this on CPU memory problem is coming , how can I avoid this, any modifications has to do, please explain

yuval-alaluf commented 1 year ago

There are already numerous issues with this same questions so please refer to them. As mentioned in the README, this code is designed to work with CUDA and we do not plan on supporting CPU-only. Unfortunately, I do not know what changes need to be made but will warn you that getting this to run on CPU can be a bit challenging because of StyleGAN's dependency on ninja.

kasarampavan commented 1 year ago

Thank you so much for your reply