Separation of Pre-Procession Diffusion block from the model

[satijapratik]

Hey,

The approach mentioned in this paper seems very intriguing and useful. However, I wanted to test out the usage of just the pre-processing phase where the key points are repaired. Are there any particular steps that I can be pointed towards so that I am able to just observe the broken and repaired key points after observation is passed through the Pre-Process Diffusion block?