-
Notifications
You must be signed in to change notification settings - Fork 5.8k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
adapt PixArtAlphaPipeline for pixart-lcm model #5974
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Superb! Let's add a test too? @patrickvonplaten WDYT?
Would be very interesting to see some results with timing info.
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
@lawrence-cj let's maybe add some results in here? Also, would be great to mention this from the PixArt docs and add a code snippet. |
@sayakpaul No problem, I will make it done asap |
@sayakpaul feel free to merge once you're happy with it |
Great job @lawrence-cj ! |
So, we are waiting for the docs here. |
The model card is updated: https://huggingface.co/PixArt-alpha/PixArt-LCM-XL-2-1024-MS |
* adapt PixArtAlphaPipeline for pixart-lcm model * remove original_inference_steps from __call__ --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
* adapt PixArtAlphaPipeline for pixart-lcm model * remove original_inference_steps from __call__ --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
What does this PR do?
Adapt the PixArtAlphaPipeline for PixArt-lcm model. Cc: @sayakpaul
Adding the retrieve_timesteps() for LCMScheduler usage.