Skip to content

Commit

Permalink
image_url -> url
Browse files Browse the repository at this point in the history
  • Loading branch information
sayakpaul committed Dec 4, 2023
1 parent 1782d5a commit df5dead
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions benchmarks/base_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,10 +142,10 @@ class ControlNetBenchmark(TextToImageBenchmark):
pipeline_class = StableDiffusionControlNetPipeline
aux_network_class = ControlNetModel

image_url = (
url = (
"https://huggingface.co/datasets/diffusers/docs-images/resolve/main/benchmarking/canny_image_condition.png"
)
image = load_image(image_url).convert("RGB")
image = load_image(url).convert("RGB")

def __init__(self, args):
if isinstance(self.pipeline_class, StableDiffusionControlNetPipeline):
Expand Down

0 comments on commit df5dead

Please sign in to comment.