guohp123456 commited on
Commit
edd07b9
·
verified ·
1 Parent(s): 391b05f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from utils import generate_canny
7
  # 加载模型
8
  pipe = DiffusionPipeline.from_pretrained(
9
  "stabilityai/stable-diffusion-3.5-large-controlnet-canny",
 
10
  torch_dtype=torch.float16,
11
  variant="fp16"
12
  ).to("cuda")
 
7
  # 加载模型
8
  pipe = DiffusionPipeline.from_pretrained(
9
  "stabilityai/stable-diffusion-3.5-large-controlnet-canny",
10
+
11
  torch_dtype=torch.float16,
12
  variant="fp16"
13
  ).to("cuda")