I had a lot of problems getting this to work and I do have some other ideas on how I want to use the denoising step. Thus, I decided to make a fork of my own. While it needs a bit more work to get started it should be pretty easy for anyone with a bit of experience with python venvs and pytorch. Feel free to give it a go and let me know if you have any issues.