controlnet_depth.py contains the core implementations of Modulate&Gate module
vae.py contains the core implementation of the Fidelity Encoder
train_depth.yaml contains the settings of our DiffUR^3
train_depth.py
inference_depth.py 

Note that we use depth to refer to stage I, because we initially intended to incorporate depth information as the condition.