torchdyn>=1.0.3 breaks if a model does not have *args, **kwargs in the forward method. This breaks downstream code which has custom models compatible with previous torchdyn versions. I propose to remove this requirement.
atong01/conditional-flow-matching#60