search

atong01 / conditional-flow-matching

TorchCFM: a Conditional Flow Matching library
https://arxiv.org/abs/2302.00482
MIT License
1.25k stars 101 forks source link

sigma should be a float #56

Closed kilianFatras closed 1 year ago

kilianFatras commented 1 year ago

There is a bug when we set sigma to be an integer (FM = ConditionalFlowMatcher(sigma=0)). It modifies the type of the time tensor t.

kilianFatras commented 1 year ago

Fixed with #72