mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization
MIT License
1.42k stars 206 forks source link

input must be a CUDA tensor #41

Closed samwang90 closed 2 years ago

samwang90 commented 2 years ago

when set device to cpu in Colab, e4e_projection gives error "input must be a CUDA tensor"