Spaces:
Runtime error
Runtime error
fix pipeline
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ conditioning_image_transforms = T.Compose(
|
|
19 |
)
|
20 |
|
21 |
cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True)
|
22 |
-
pipe =
|
23 |
"./models/wd-1-5-b2",
|
24 |
controlnet=cnet,
|
25 |
torch_dtype=torch.float16,
|
|
|
19 |
)
|
20 |
|
21 |
cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True)
|
22 |
+
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
23 |
"./models/wd-1-5-b2",
|
24 |
controlnet=cnet,
|
25 |
torch_dtype=torch.float16,
|