From c05356497a07fa1ffb1cc95657e7aa5d0606ec5f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 30 Nov 2022 13:57:45 +0100 Subject: [PATCH] Add better docs xformers (#1487) * Add better docs xformers * update * Apply suggestions from code review * fix --- docs/source/api/pipelines/stable_diffusion.mdx | 11 ++++++++++- docs/source/using-diffusers/loading.mdx | 18 ------------------ 2 files changed, 10 insertions(+), 19 deletions(-) diff --git a/docs/source/api/pipelines/stable_diffusion.mdx b/docs/source/api/pipelines/stable_diffusion.mdx index 70c4abaaf6..6068b961ae 100644 --- a/docs/source/api/pipelines/stable_diffusion.mdx +++ b/docs/source/api/pipelines/stable_diffusion.mdx @@ -78,25 +78,32 @@ If you want to use all possible use cases in a single `DiffusionPipeline` you ca - disable_attention_slicing - enable_vae_slicing - disable_vae_slicing + - enable_xformers_memory_efficient_attention + - disable_xformers_memory_efficient_attention ## StableDiffusionImg2ImgPipeline [[autodoc]] StableDiffusionImg2ImgPipeline - __call__ - enable_attention_slicing - disable_attention_slicing + - enable_xformers_memory_efficient_attention + - disable_xformers_memory_efficient_attention ## StableDiffusionInpaintPipeline [[autodoc]] StableDiffusionInpaintPipeline - __call__ - enable_attention_slicing - disable_attention_slicing - + - enable_xformers_memory_efficient_attention + - disable_xformers_memory_efficient_attention ## StableDiffusionImageVariationPipeline [[autodoc]] StableDiffusionImageVariationPipeline - __call__ - enable_attention_slicing - disable_attention_slicing + - enable_xformers_memory_efficient_attention + - disable_xformers_memory_efficient_attention ## StableDiffusionUpscalePipeline @@ -104,3 +111,5 @@ If you want to use all possible use cases in a single `DiffusionPipeline` you ca - __call__ - enable_attention_slicing - disable_attention_slicing + - enable_xformers_memory_efficient_attention + - disable_xformers_memory_efficient_attention diff --git a/docs/source/using-diffusers/loading.mdx b/docs/source/using-diffusers/loading.mdx index c97ad5c5d0..97bb7a0d03 100644 --- a/docs/source/using-diffusers/loading.mdx +++ b/docs/source/using-diffusers/loading.mdx @@ -378,21 +378,3 @@ dpm = DPMSolverMultistepScheduler.from_pretrained(repo_id, subfolder="scheduler" # replace `dpm` with any of `ddpm`, `ddim`, `pndm`, `lms`, `euler`, `euler_anc` pipeline = StableDiffusionPipeline.from_pretrained(repo_id, scheduler=dpm) ``` - -## API - -[[autodoc]] modeling_utils.ModelMixin - - from_pretrained - - save_pretrained - -[[autodoc]] pipeline_utils.DiffusionPipeline - - from_pretrained - - save_pretrained - -[[autodoc]] modeling_flax_utils.FlaxModelMixin - - from_pretrained - - save_pretrained - -[[autodoc]] pipeline_flax_utils.FlaxDiffusionPipeline - - from_pretrained - - save_pretrained