Code in this folder except in the `BRGM` subfolder is taken from there: https://huggingface.co/spaces/Warvito/diffusion_brain/tree/main

The code in the `BRGM` subfolder is from: 
https://github.com/razvanmarinescu/brgm-pytorch