ts: callback(step: int, timestep: int, latents: torch.FloatTensor).


callback_steps (int, optional, defaults to 1) —
The frequency at which the callback function will be called. If not specified, the callback will be
called at every step.


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 o