Code-Cooker

Sleeping

Severian commited on Sep 6, 2024

Commit

18089e1

verified ·

1 Parent(s): a46dff1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,11 +29,11 @@ qrcode_generator = qrcode.QRCode(
 )
 controlnet = ControlNetModel.from_pretrained(
-    "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
 ).to("cuda")
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
     controlnet=controlnet,
     safety_checker=None,
     torch_dtype=torch.float16,

 )
 controlnet = ControlNetModel.from_pretrained(
+    "monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16
 ).to("cuda")
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
+    "rupeshs/LCM-runwayml-stable-diffusion-v1-5",
     controlnet=controlnet,
     safety_checker=None,
     torch_dtype=torch.float16,