Does the model work with PyTorch v2.x?

e4s2022 / e4s

(CVPR 2023) E4S: Fine-grained Face Swapping via Regional GAN Inversion

https://e4s2022.github.io

MIT License

368 stars 32 forks source link

Does the model work with PyTorch v2.x? #38

Open jackmvision opened 3 months ago

jackmvision commented 3 months ago

Hi, I was trying to run the demo with Pytorch v2.x. It seems that the swapped face is black. When i switched the platform back to v1.x, it works. Anyone had the same problem? Any suggestion to make this work with PT v2.x?

Thanks!

vijeshkpaei commented 2 months ago

Hi @jackmvision , 2.x working !! torch == 2.0.0 torch-fidelity==0.3.0 torchmetrics ==1.4.0.post0 torchvision==0.14.1

jackmvision commented 2 months ago

Thanks @vijeshkpaei ! Will give a try