sayakpaul HF staff commited on
Commit
8342365
1 Parent(s): 053606a

fix: image url.

Browse files
Files changed (1) hide show
  1. utils.py +4 -4
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-hed/resolve/main/images/bird.png"
33
- )
34
  elif pipeline_to_benchmark == "SDXL ControlNet":
35
  image = load_image(
36
- "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/hf-logo.png"
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