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
-
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