From a042458163c95049f051545c940722be8b712f97 Mon Sep 17 00:00:00 2001 From: Michael Katsoulis Date: Tue, 21 Nov 2023 10:01:21 +0000 Subject: [PATCH] Controlnet logo to gpu --- examples/logo-controlnet.mdx | 1 + 1 file changed, 1 insertion(+) 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