Update controlnet_demo.py

hutaiHang · Dec 21, 2023 · d3b6800 · d3b6800
1 parent 8307ec9
commit d3b6800
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/controlnet_demo.py b/controlnet_demo.py
@@ -22,7 +22,7 @@
 image_condition = Image.fromarray(image)
 
 controlnet = ControlNetModel.from_pretrained(
-    "/home/u2120230624/share/ControlNet/canny", torch_dtype=torch.float16
+    "lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16
 ).to('cuda')
 
 pipe = StableDiffusionControlNetPipeline.from_pretrained(