Skip to content

GitLab

Unverified Commit b8ccb462 authored by Tolga Cangöz's avatar Tolga Cangöz Committed by GitHub
Browse files

Fix CPU offload in docstring (#7827) 

Fix cpu offload
parent 725ead2f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py
View file @ b8ccb462
......@@ -89,8 +89,8 @@ EXAMPLE_DOC_STRING = """
... variant="fp16",
... use_safetensors=True,
... torch_dtype=torch.float16,
... ).to("cuda")
>>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16).to("cuda")
... )
>>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
>>> pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
... "stabilityai/stable-diffusion-xl-base-1.0",
... controlnet=controlnet,
......@@ -98,7 +98,7 @@ EXAMPLE_DOC_STRING = """
... variant="fp16",
... use_safetensors=True,
... torch_dtype=torch.float16,
... ).to("cuda")
... )
>>> 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