Spaces:
Paused
Paused
Commit
·
5768419
1
Parent(s):
e1ed3fd
cpu load
Browse files
app.py
CHANGED
@@ -30,14 +30,14 @@ 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",
|
37 |
controlnet=controlnet,
|
38 |
safety_checker=None,
|
39 |
torch_dtype=torch.float16,
|
40 |
-
)
|
41 |
# pipe.enable_xformers_memory_efficient_attention()
|
42 |
|
43 |
|
|
|
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",
|
37 |
controlnet=controlnet,
|
38 |
safety_checker=None,
|
39 |
torch_dtype=torch.float16,
|
40 |
+
) #.to("cuda")
|
41 |
# pipe.enable_xformers_memory_efficient_attention()
|
42 |
|
43 |
|