diffengine.datasets.utils

Module Contents

Functions

encode_prompt_sdxl(batch, text_encoders, tokenizers, ...)

Encode prompt for Stable Diffusion XL.

diffengine.datasets.utils.encode_prompt_sdxl(batch, text_encoders, tokenizers, caption_column, proportion_empty_prompts=0.0, *, is_train=True)

Encode prompt for Stable Diffusion XL.

Parameters:

• proportion_empty_prompts (float) –

• is_train (bool) –

Return type:

dict[str, torch.Tensor]