Error when exporting FrEIA library to ONNX

[Tekno-H]

I am trying to export the CFlow-ad model to ONNX, however the export always fails at line 248 in the all_in_one_block.py file.
No error message is displayed, only a graph dump happens.
Any ideas what may be causing this problem?
This is an issue I opened in PyTorch repo.

[dibetgonzalez]

Hi, do you have any advanced in this issue?