YangLing0818 / consistency_flow_matching

Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
https://arxiv.org/abs/2407.02398
131 stars 4 forks source link

Sampling via re-noise similar to consistency model? #2

Open XavierXiao opened 2 months ago

XavierXiao commented 2 months ago

Thanks for the great work and code release! I have one question: assuming we have only one segment for simplicity, in the paper you mentioned that Euler solver is used to sample from the model. But since it is partially trained with the consistency objective, can we adopt the sampling method in consistency model (adding noise to predicted x_0)? Thanks!