 prior_callback_on_step_end(self: DiffusionPipeline, step: int, timestep: int, callback_kwargs: Dict).   prior_callback_on_step_end_tensor_inputs (List, optional) —
The list of tensor inputs for the prior_callback_on_step_end function. The tensors specified in the
list will be passed as callback_kwargs argument. You will only be able to include variables listed in
the ._callback_tensor_inputs attribute of your pipeine class.   callback_on_step_