rch.Tensor).   return_dict (bool, optional, defaults to True) —
Whether or not to return a ImagePipelineOutput instead of a plain tuple.   prior_callback_on_step_end (Callable, optional) —
A function that calls at the end of each denoising steps during the inference. The function is called
with the following arguments: prior_callback_on_step_end(self: DiffusionPipeline, step: int, timestep: int, callback_kwargs: Dict).   prior_callback_on_step_