It is a minimal repro of my project.
Click here to download the pytorch model 'gen.pth'. Then put it into ./models .
My version has been given In the comments above. Just run the file viton-with-HR.ipynb in this environment.
Thank you very much for helping me with my problem!