here is a bug

[Johnson-yue]

https://github.com/VLL-HD/conditional_invertible_neural_networks/blob/45dc7250ebfecf10d1a278edafde0fe899f30aa1/colorization_cINN/model.py#L64

Hi ,cINN code have split here, but some new question .
FrEIA has updated to version 0.2.0 ????
I see some new function or usage in colorization_cINN.model such as subnet_coupling_layer, glow_coupling_layer, conv_1x1 .

Can you update the Docs and Tutorial for this version?