(step: int, timestep: int, latents: torch.FloatTensor).   callback_steps (int, optional, defaults to 1) —
The frequency at which the callback function is called. If not specified, the callback is called at
every step.
extra_step_kwargs —
Extra_step_kwargs.
cross_attention_kwargs —
A kwargs dictionary that if specified is passed along to the AttentionProcessor as defined in
self.processor.
num_warmup_steps —
number of warmup steps.   Returns 
la