 listed in the
._callback_tensor_inputs attribute of your pipeline class.   Returns 
ImagePipelineOutput or tuple
    Function invoked when calling the pipeline for generation.  Examples:   Copied from diffusers import AutoPipelineForText2Image
import torch

pipe = AutoPipelineForText2Image.from_pretrained(
    "kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16
)
pipe.enable_model_cpu_offload()

prompt = "A lion in galaxies,