Training with flow-matching

Hello!
I'm apperciate for your job. I am very interested in PTX-mini and want to train this model with flow matching. Unfortunately, I can not find the corresponding code in this git. Can you relase flow matching version training code?
best wishes