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