diff --git a/examples/logo-controlnet.mdx b/examples/logo-controlnet.mdx index d5ecf012..b2725690 100644 --- a/examples/logo-controlnet.mdx +++ b/examples/logo-controlnet.mdx @@ -78,6 +78,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained( torch_dtype=torch.float16, ) pipe.enable_model_cpu_offload() +pipe = pipe.to("cuda") ``` ## Predict Function