Uh oh!
There was an error while loading. Please reload this page.
[lavender-flow] use flow match euler scheduler - #8799
Conversation
HuggingFaceDocBuilderDev
commented
Jul 5, 2024
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
| # 4. Prepare timesteps | ||
| sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps) |
There was a problem hiding this comment.
I added this just to get 1:1 with the original implementation, I don't think it's needed here, the difference should be very small, but I didn't test for smaller number of steps
cc @cloneofsimo
sayakpaul
left a comment
There was a problem hiding this comment.
Thanks. Just a single comment.
| torch.tensor([t / 1000]) | ||
| .expand(latent_model_input.shape[0]) | ||
| .to(latents.device, dtype=latents.dtype) |
There was a problem hiding this comment.
We could unpack this code a bit and perhaps add a comment on why we're dividing by 1000. This is mainly because we're moving away from our normal pipeline implementations a bit which our users are used to reading and referring to.
There was a problem hiding this comment.
done! feel free to merge into your PR now
sayakpaul
commented
Jul 7, 2024
Thanks a lot for your help, Yiyi! |


part of #8796
you can use this to test once this PR is fixed#8798