Skip to content

GitLab

Unverified Commit 714bfed8 authored by Steven Liu's avatar Steven Liu Committed by GitHub
Browse files

[docs] Fix ControlNet SDXL docstring (#4582) 

fix
parent d5983a67
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py
View file @ 714bfed8
...@@ -81,8 +81,8 @@ EXAMPLE_DOC_STRING = """ ...@@ -81,8 +81,8 @@ EXAMPLE_DOC_STRING = """
... "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16 ... "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16
... ) ... )
>>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16) >>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
>>> pipe = StableDiffusionControlNetPipeline.from_pretrained( >>> pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
... "stabilityai/stable-diffusion-xl-base-1.0", controlnet=controlnet, torch_dtype=torch.float16 ... "stabilityai/stable-diffusion-xl-base-1.0", controlnet=controlnet, vae=vae, torch_dtype=torch.float16
... ) ... )
>>> pipe.enable_model_cpu_offload() >>> pipe.enable_model_cpu_offload()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment