torchdyn
torchdyn copied to clipboard
Model require `forward` to have `*args, **kwargs` in 1.0.6
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.
https://github.com/atong01/conditional-flow-matching/issues/60