how to use inference.py with pretrained model on cpu

omertov / encoder4editing

Official implementation of "Designing an Encoder for StyleGAN Image Manipulation" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02766

MIT License

945 stars 154 forks source link

how to use inference.py with pretrained model on cpu #66

Open jeffryWillam opened 2 years ago

jeffryWillam commented 2 years ago

Thanks for your great work! The results are amazing! However, I found when I only use the inference.py with pretrained model , it still needs a GPU there, is it possible to use only the cpu for inference with pretrained model, as this will be more flexible. Thanks!!!