The number of function evaluation (NFE) for SiT

shiml20 / FlowTurbo

Official PyTorch Implementation of "FlowTurbo: Towards Real-time Flow-Based Image Generation with Velocity Refiner"

https://flowturbo.ivg-research.xyz/

MIT License

59 stars 3 forks source link

The number of function evaluation (NFE) for SiT #5

Open dlsrbgg33 opened 3 weeks ago

dlsrbgg33 commented 3 weeks ago

Hi, thank you for your awesome work.

I have a question regarding the NFEs used in SiT. The original SiT model employed 125 steps of Heun's method, resulting in 250 NFEs. However, you mentioned that 25 steps (50 NFEs) can achieve a similar score in the ODE setting. Could you provide some insight into why this might be the case?

Thank you in advance.