Hugging Face's logo

Spaces:
diffusers
/
benchmark-pt2.1
Sleeping

App Files Community
benchmark-pt2.1
File size: 2,604 Bytes 
57f479a
 
 
 
 
 
 
fd14e0a
57f479a
 
 
 
 
 
7690ae5
 
57f479a
 
 
 
7690ae5
57f479a
 
 
 
 
 
 
 
 
 
8342365
 
57f479a
 
cf42f63
8342365
57f479a
 
 
 
 
 
 
84892c9
 
57f479a
 
 
09cae90
57f479a
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
from diffusers.utils import load_image


def get_image_for_img_to_img(pipeline_to_benchmark):
    if pipeline_to_benchmark == "SD I2I":
        url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/assets/stable-samples/img2img/sketch-mountains-input.jpg"
        init_image = load_image(url).convert("RGB")
        init_image = init_image.resize((512, 512))
    elif pipeline_to_benchmark == "SDXL I2I":
        url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/sdxl-img2img.png"
        init_image = load_image(url).convert("RGB")
    return init_image


def get_image_for_inpainting(pipeline_to_benchmark):
    if pipeline_to_benchmark == "SD Inpainting":
        image_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
        mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
        init_image = load_image(image_url).convert("RGB").resize((512, 512))
        mask_image = load_image(mask_url).convert("RGB").resize((512, 512))
    elif pipeline_to_benchmark == "SDXL Inpainting":
        image_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/sdxl-text2img.png"
        mask_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/sdxl-inpaint-mask.png"
        init_image = load_image(image_url).convert("RGB")
        mask_image = load_image(mask_url).convert("RGB")
    return init_image, mask_image


def get_image_for_controlnet(pipeline_to_benchmark):
    if pipeline_to_benchmark == "SD ControlNet":
        image = load_image(
            "https://huggingface.co/lllyasviel/sd-controlnet-canny/resolve/main/images/bird_canny.png"
        ).convert("RGB")
    elif pipeline_to_benchmark == "SDXL ControlNet":
        image = load_image(
            "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/hf-logo.png"
        ).convert("RGB")

    return image


def get_image_for_adapters(pipeline_to_benchmark):
    if pipeline_to_benchmark == "SD T2I Adapters":
        image = load_image(
            "https://huggingface.co/datasets/sayakpaul/sample-datasets/resolve/main/sd_t2i_canny.png"
        )
    elif pipeline_to_benchmark == "SDXL T2I Adapters":
        image = load_image(
            "https://huggingface.co/Adapter/t2iadapter/resolve/main/figs_SDXLV1.0/cond_canny.png"
        ).convert("RGB")
    return image