diffengine.datasets.hf_esd_datasets

Module Contents

Classes

HFESDDatasetPreComputeEmbs

Huggingface Erasing Concepts from Diffusion Models Dataset.

class diffengine.datasets.hf_esd_datasets.HFESDDatasetPreComputeEmbs(forget_caption, model='stabilityai/stable-diffusion-xl-base-1.0', device='cuda', pipeline=())

Bases: torch.utils.data.Dataset

Huggingface Erasing Concepts from Diffusion Models Dataset.

Dataset of huggingface datasets for Erasing Concepts from Diffusion Models.

Args:

forget_caption (str): The caption used to forget. model (str): pretrained model name of stable diffusion xl.

Defaults to ‘stabilityai/stable-diffusion-xl-base-1.0’.

device (str): Device used to compute embeddings. Defaults to ‘cuda’. pipeline (Sequence): Processing pipeline. Defaults to an empty tuple.

__len__()

Get the length of dataset.

Returns:

int

Return type:

The length of filtered dataset.

__getitem__(idx)

Get the dataset after ``self.pipeline`.

Args:

idx (int): The index.

Returns:

dict: The idx-th data information of dataset after self.pipeline.

Parameters:

idx (int) –

Return type:

dict

Parameters:

• forget_caption (str) –

• model (str) –

• device (str) –

• pipeline (collections.abc.Sequence) –