Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|