Controlnet logo to gpu

CerebriumAI · Nov 21, 2023 · a042458 · a042458
1 parent 0253324
commit a042458
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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