xywwww commited on
Commit
370c493
·
verified ·
1 Parent(s): dd1e182

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ from PIL import Image
69
  import numpy as np
70
 
71
  # 初始化 ControlNet 模型和 Stable Diffusion Pipeline
72
- controlnet = ControlNetModel.from_pretrained("controlnet/checkpoints/ControlNetModel")
73
  pipeline = StableDiffusionControlNetPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", controlnet=controlnet).to("cuda")
74
 
75
  # 定义图像生成函数
 
69
  import numpy as np
70
 
71
  # 初始化 ControlNet 模型和 Stable Diffusion Pipeline
72
+ controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
73
  pipeline = StableDiffusionControlNetPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", controlnet=controlnet).to("cuda")
74
 
75
  # 定义图像生成函数