radames commited on
Commit
a46dff1
1 Parent(s): 3e16b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ qrcode_generator = qrcode.QRCode(
30
 
31
  controlnet = ControlNetModel.from_pretrained(
32
  "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
33
- )
34
 
35
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
36
  "runwayml/stable-diffusion-v1-5",
 
30
 
31
  controlnet = ControlNetModel.from_pretrained(
32
  "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
33
+ ).to("cuda")
34
 
35
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
36
  "runwayml/stable-diffusion-v1-5",