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.   clean_caption (bool, optional, defaults to True) —
Whether or not to clean the caption before creating embeddings. Requires beautifulsoup4 and ftfy to
be installed. If the dependencies are not installed, th