Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/diffusers.git" did not exist on "fa31da29e591ed2e64a7c6ba9153c0b2e5a0ddc2"
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