* fix a bug in flow matching * fixed the bugs in flow matching * fixed format --------- Co-authored-by: Geoffrey19 <sympathischmann35@gmail.com>