TencentARC / PhotoMaker

PhotoMaker
https://photo-maker.github.io/
Other
8.63k stars 675 forks source link

add capability to interrupt the pipeline #119

Closed machineminded closed 5 months ago

machineminded commented 5 months ago

This PR adds the ability to interrupt the pipeline by passing a callback_on_step_end argument per documentation here:

https://huggingface.co/docs/diffusers/en/using-diffusers/callback#interrupt-the-diffusion-process

The example usage provided in the URL above should work as it is what I am using in Fooocus:

https://github.com/machineminded/Fooocus-inswapper/blob/main/modules/pm.py#L22 https://github.com/machineminded/Fooocus-inswapper/blob/main/modules/pm.py#L131

Paper99 commented 5 months ago

Thanks for your great work! We've merged your PR.