Versatile Diffusion

Versatile Diffusion was proposed in Versatile Diffusion: Text, Images and Variations All in One Diffusion Model by Xingqian Xu, Zhangyang Wang, Eric Zhang, Kai Wang, Humphrey Shi .

The abstract from the paper is:

The recent advances in diffusion models have set an impressive milestone in many generation tasks. Trending works such as DALL-E2, Imagen, and Stable Diffusion have attracted great interest in academia and industry. Despite the rapid landscape changes, recent new approaches focus on extensions and performance rather than capacity, thus requiring separate models for separate tasks. In this work, we expand the existing single-flow diffusion pipeline into a multi-flow network, dubbed Versatile Diffusion (VD), that handles text-to-image, image-to-text, image-variation, and text-variation in one unified model. Moreover, we generalize VD to a unified multi-flow multimodal diffusion framework with grouped layers, swappable streams, and other propositions that can process modalities beyond images and text. Through our experiments, we demonstrate that VD and its underlying framework have the following merits: a) VD handles all subtasks with competitive quality; b) VD initiates novel extensions and applications such as disentanglement of style and semantic, image-text dual-guided generation, etc.; c) Through these experiments and applications, VD provides more semantic insights of the generated outputs.

Tips

You can load the more memory intensive "all-in-one" [VersatileDiffusionPipeline] that supports all the tasks or use the individual pipelines which are more memory efficient.

Pipeline	Supported tasks
[`VersatileDiffusionPipeline`]	all of the below
[`VersatileDiffusionTextToImagePipeline`]	text-to-image
[`VersatileDiffusionImageVariationPipeline`]	image variation
[`VersatileDiffusionDualGuidedPipeline`]	image-text dual guided generation

Make sure to check out the Schedulers guide to learn how to explore the tradeoff between scheduler speed and quality, and see the reuse components across pipelines section to learn how to efficiently load the same components into multiple pipelines.

VersatileDiffusionPipeline

[[autodoc]] VersatileDiffusionPipeline

VersatileDiffusionTextToImagePipeline

[[autodoc]] VersatileDiffusionTextToImagePipeline - all - call

VersatileDiffusionImageVariationPipeline

[[autodoc]] VersatileDiffusionImageVariationPipeline - all - call

VersatileDiffusionDualGuidedPipeline

[[autodoc]] VersatileDiffusionDualGuidedPipeline - all - call