Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|