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 = """ ...@@ -89,8 +89,8 @@ EXAMPLE_DOC_STRING = """
... variant="fp16", ... variant="fp16",
... use_safetensors=True, ... use_safetensors=True,
... torch_dtype=torch.float16, ... 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( >>> pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
... "stabilityai/stable-diffusion-xl-base-1.0", ... "stabilityai/stable-diffusion-xl-base-1.0",
... controlnet=controlnet, ... controlnet=controlnet,
...@@ -98,7 +98,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -98,7 +98,7 @@ EXAMPLE_DOC_STRING = """
... variant="fp16", ... variant="fp16",
... use_safetensors=True, ... use_safetensors=True,
... torch_dtype=torch.float16, ... torch_dtype=torch.float16,
... ).to("cuda") ... )
>>> 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