arguments in callback function restricted to ["latents", "prompt_embeds", "negative_prompt_embeds"] in pipeline_stable_diffusion.py #8311
Varghese-Kuruvilla
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Why are the arguments in the pipeline callback restricted to ["latents", "prompt_embeds", "negative_prompt_embeds"]?
Line 176 (stable diffusion.py)
_callback_tensor_inputs = ["latents", "prompt_embeds", "negative_prompt_embeds"]
How can I access other variables, for example 'timesteps' without changing the source code? Not sure if I had to put this as a discussion or its better if its moved to issues. Thanks in advance!
Beta Was this translation helpful? Give feedback.
All reactions