Subclass torch's LayerNorm to handle fp16.

A dataset to prepare the instance and class images with the prompts for fine-tuning the model. It pre-processes the images and the tokenizes

Uses the CLIP transformer encoder for text (from Hugging Face)

Apply gaussian smoothing on a 1d, 2d or 3d tensor. Filtering is performed seperately for each channel in the input using a depthwise conv

A ResNet class that is similar to torchvision's but contains the following changes: - There are now 3 "stem" convolutions as opposed to 1, wi

r"""Allows the model to jointly attend to information from different representation subspaces. See reference: Attention Is All You Need .

A simple dataset to prepare the prompts to generate class images on multiple GPUs.

Wraps an arbitrary object with __len__ and __getitem__ into a pytorch dataset

Apply average smoothing on a 1d, 2d or 3d tensor. Filtering is performed seperately for each channel in the input using a depthwise convo

r""" Pipeline for custom diffusion model. This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic