ar0551 commited on
Commit
13e92ca
·
verified ·
1 Parent(s): 144d877

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -34,6 +34,8 @@ pipe_cn = StableDiffusionXLControlNetPipeline.from_pretrained(
34
  scheduler=eulera_scheduler,
35
  )
36
 
 
 
37
  # Stable Diffusion Model without ControlNet
38
  pipe = StableDiffusionXLPipeline.from_pretrained(
39
  "stabilityai/stable-diffusion-xl-base-1.0",
 
34
  scheduler=eulera_scheduler,
35
  )
36
 
37
+ pipe_cn.to(device)
38
+
39
  # Stable Diffusion Model without ControlNet
40
  pipe = StableDiffusionXLPipeline.from_pretrained(
41
  "stabilityai/stable-diffusion-xl-base-1.0",