update

meanaudio/model/flow_matching.py

@@ -66,6 +66,8 @@ class FlowMatching:
                 flow = fn(t, x)
                 next_t = steps[ti + 1]
                 dt = next_t - t
-                x = x + dt * flow
+
+                # fix: we need to subtract the flow since we learn the reverse trajectory
+                x = x - dt * flow
 
         return x