API deprecation.

[jrodriguezpuigvert]

Hi, thanks for sharing your amazing work.
I think that you change the api implementation and not the call. Easy to fix.

https://github.com/VLL-HD/conditional_INNs/blob/19e316c606cae24815efa51305ce8d3a6476f819/mnist_minimal_example/model.py#L57

**
z ,jac = self.cinn(x, c=one_hot(l))**