ch the callback function is called. If not specified, the callback is called at
every step.   cross_attention_kwargs (dict, optional) —
A kwargs dictionary that if specified is passed along to the
AttnProcessor as defined in
self.processor.   lambda_auto_corr (float, optional, defaults to 20.0) —
Lambda parameter to control auto correction.   lambda_kl (float, optional, defaults to 20.0) —
Lambda parameter to control Kullback-Leibler divergence