File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/diffusers/pipelines/controlnet Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 89
89
... variant="fp16",
90
90
... use_safetensors=True,
91
91
... torch_dtype=torch.float16,
92
- ... ).to("cuda")
93
- >>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16).to("cuda")
92
+ ... )
93
+ >>> vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
94
94
>>> pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
95
95
... "stabilityai/stable-diffusion-xl-base-1.0",
96
96
... controlnet=controlnet,
97
97
... vae=vae,
98
98
... variant="fp16",
99
99
... use_safetensors=True,
100
100
... torch_dtype=torch.float16,
101
- ... ).to("cuda")
101
+ ... )
102
102
>>> pipe.enable_model_cpu_offload()
103
103
104
104
You can’t perform that action at this time.
0 commit comments