Spaces:
Running
on
A10G
Running
on
A10G
fix: image url.
Browse files
utils.py
CHANGED
@@ -29,12 +29,12 @@ def get_image_from_inpainting(pipeline_to_benchmark):
|
|
29 |
def get_image_for_controlnet(pipeline_to_benchmark):
|
30 |
if pipeline_to_benchmark == "SD ControlNet":
|
31 |
image = load_image(
|
32 |
-
"https://huggingface.co/lllyasviel/sd-controlnet-
|
33 |
-
)
|
34 |
elif pipeline_to_benchmark == "SDXL ControlNet":
|
35 |
image = load_image(
|
36 |
-
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/
|
37 |
-
)
|
38 |
|
39 |
return image
|
40 |
|
|
|
29 |
def get_image_for_controlnet(pipeline_to_benchmark):
|
30 |
if pipeline_to_benchmark == "SD ControlNet":
|
31 |
image = load_image(
|
32 |
+
"https://huggingface.co/lllyasviel/sd-controlnet-canny/resolve/main/images/bird_canny.png"
|
33 |
+
).convert("RGB")
|
34 |
elif pipeline_to_benchmark == "SDXL ControlNet":
|
35 |
image = load_image(
|
36 |
+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/blob/main/sd_controlnet/hf-logo.png"
|
37 |
+
).convert("RGB")
|
38 |
|
39 |
return image
|
40 |
|