Question about one-step inference

[wangjg33]

Great job!
But we find that num_inreference_steps did not work and remained a 4-step inference.
Upon checking the code, it was found that timestep was forcibly set here.. How can I test the results under 1step conditions.

[CSRuiXie]

Thank you for your attention. The code is hard-coded, sorry for the misunderstanding. You can change the inference step here to use a different step (e.g., if you want one-step inference, set 't == 999').

[Chenglu0426]

Hello, if I want to inference the model for 10 steps, how to modify the code?