Types & classes — github.com/adobe-research/custom-diffusion

CodeHub MCPcopy Create free account

hub / github.com/adobe-research/custom-diffusion / types & classes

Types & classes19 in github.com/adobe-research/custom-diffusion

⨍ Functions 135◇ Types & classes 19

↓ 3 callersClassCustomDiffusionAttnProcessor

src/diffusers_model_pipeline.py:279

↓ 2 callersClassCustomDiffusionDataset

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

src/diffusers_data_pipeline.py:275

↓ 2 callersClassPromptDataset

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

src/diffusers_data_pipeline.py:258

↓ 1 callersClassCLIPCapDataset

customconcept101/evaluate.py:19

↓ 1 callersClassConcatDataset

↓ 1 callersClassCustomDiffusionXFormersAttnProcessor

src/diffusers_model_pipeline.py:323

↓ 1 callersClassFrozenCLIPEmbedderWrapper

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

src/custom_modules.py:226

↓ 1 callersClassWrappedDataset

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

ClassAbstractEncoder

src/custom_modules.py:218

ClassCLIPImageDataset

customconcept101/evaluate.py:41

ClassCUDACallback

ClassCustomDiffusion

src/model.py:106

ClassCustomDiffusionPipeline

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

src/diffusers_model_pipeline.py:367

ClassCustomDiffusionXLPipeline

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

src/diffusers_model_pipeline.py:503

ClassDINOImageDataset

customconcept101/evaluate.py:67

ClassDataModuleFromConfig

ClassImageLogger

src/finetune_data.py:115

ClassSetupCallback