From cbc4166f1086b9c47e61a12f41222ee8a4a90463 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 9 May 2023 21:51:17 -0500 Subject: [PATCH] fix(scripts): add more release tests --- .../outpaint-panorama-even-256-0.png | 3 + .../outpaint-panorama-horizontal-512-0.png | 3 + .../outpaint-panorama-vertical-512-0.png | 3 + .../txt2img-sd-v1-5-512-muffin-cloud-0.png | 3 + .../txt2img-sd-v1-5-512-muffin-taters-0.png | 3 + .../upscale-codeformer-512-muffin-0.png | 3 + .../test-refs/upscale-gfpgan-muffin-0.png | 3 + .../test-refs/upscale-sd-x4-2048-muffin-0.png | 3 + .../upscale-swinir-x4-2048-muffin-0.png | 3 + api/scripts/test-release.py | 61 +++++++++++++++++++ 10 files changed, 88 insertions(+) create mode 100644 api/scripts/test-refs/outpaint-panorama-even-256-0.png create mode 100644 api/scripts/test-refs/outpaint-panorama-horizontal-512-0.png create mode 100644 api/scripts/test-refs/outpaint-panorama-vertical-512-0.png create mode 100644 api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-cloud-0.png create mode 100644 api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-taters-0.png create mode 100644 api/scripts/test-refs/upscale-codeformer-512-muffin-0.png create mode 100644 api/scripts/test-refs/upscale-gfpgan-muffin-0.png create mode 100644 api/scripts/test-refs/upscale-sd-x4-2048-muffin-0.png create mode 100644 api/scripts/test-refs/upscale-swinir-x4-2048-muffin-0.png diff --git a/api/scripts/test-refs/outpaint-panorama-even-256-0.png b/api/scripts/test-refs/outpaint-panorama-even-256-0.png new file mode 100644 index 000000000..82de6b9aa --- /dev/null +++ b/api/scripts/test-refs/outpaint-panorama-even-256-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bfb8f3f049d7985c0d95d93b84ce9d612c055b34131d23f8fe7c9868d98c456e +size 1866247 diff --git a/api/scripts/test-refs/outpaint-panorama-horizontal-512-0.png b/api/scripts/test-refs/outpaint-panorama-horizontal-512-0.png new file mode 100644 index 000000000..56761cfc4 --- /dev/null +++ b/api/scripts/test-refs/outpaint-panorama-horizontal-512-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:151c811488c933d9b858b70157da49ef626e2ce99207120fddc3791238e5a065 +size 1949395 diff --git a/api/scripts/test-refs/outpaint-panorama-vertical-512-0.png b/api/scripts/test-refs/outpaint-panorama-vertical-512-0.png new file mode 100644 index 000000000..f467648ac --- /dev/null +++ b/api/scripts/test-refs/outpaint-panorama-vertical-512-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2311f9f2c273065a854636113f43fbbae642792e7c7f265c8bb3cbda0e308182 +size 1894231 diff --git a/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-cloud-0.png b/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-cloud-0.png new file mode 100644 index 000000000..9a8d4339d --- /dev/null +++ b/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-cloud-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ac57885807fe2c2a1cbb17bfdc37d3bd6ffedad0e8de183e1039481a80e5fa34 +size 489179 diff --git a/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-taters-0.png b/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-taters-0.png new file mode 100644 index 000000000..4e0d79855 --- /dev/null +++ b/api/scripts/test-refs/txt2img-sd-v1-5-512-muffin-taters-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b97e71085b7166e6676a9c74590982f8f755c0c7256202a921aa0b2e70f06bd8 +size 451466 diff --git a/api/scripts/test-refs/upscale-codeformer-512-muffin-0.png b/api/scripts/test-refs/upscale-codeformer-512-muffin-0.png new file mode 100644 index 000000000..9af7aed24 --- /dev/null +++ b/api/scripts/test-refs/upscale-codeformer-512-muffin-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4106859b033cf57d0f2b08c36cff38beb5a67f6e99e2eddefad998223be4182 +size 498897 diff --git a/api/scripts/test-refs/upscale-gfpgan-muffin-0.png b/api/scripts/test-refs/upscale-gfpgan-muffin-0.png new file mode 100644 index 000000000..9af7aed24 --- /dev/null +++ b/api/scripts/test-refs/upscale-gfpgan-muffin-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4106859b033cf57d0f2b08c36cff38beb5a67f6e99e2eddefad998223be4182 +size 498897 diff --git a/api/scripts/test-refs/upscale-sd-x4-2048-muffin-0.png b/api/scripts/test-refs/upscale-sd-x4-2048-muffin-0.png new file mode 100644 index 000000000..0b713be10 --- /dev/null +++ b/api/scripts/test-refs/upscale-sd-x4-2048-muffin-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1a982c32e952f9fe1d9f0ba1b93f177f04f11b23603736ddc157ece2d45d915e +size 7313660 diff --git a/api/scripts/test-refs/upscale-swinir-x4-2048-muffin-0.png b/api/scripts/test-refs/upscale-swinir-x4-2048-muffin-0.png new file mode 100644 index 000000000..a7f8bde90 --- /dev/null +++ b/api/scripts/test-refs/upscale-swinir-x4-2048-muffin-0.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:76ecb7e234c8396d607d873a2f727164b76436e0ecb00ece65a4c38de4763a95 +size 5419174 diff --git a/api/scripts/test-release.py b/api/scripts/test-release.py index 10d069fb7..f0d7ed82d 100644 --- a/api/scripts/test-release.py +++ b/api/scripts/test-release.py @@ -193,6 +193,67 @@ def __init__( "txt2img-sd-v2-1-512-muffin-0", ], ), + TestCase( + "txt2img-sd-v1-5-512-muffin-taters", + "txt2img?prompt=+a+giant+muffin+made+of+mashed+potatoes&seed=0&scheduler=unipc-multi", + ), + TestCase( + "txt2img-sd-v1-5-512-muffin-cloud", + "txt2img?prompt=+a+giant+muffin+made+of+cloud-all&seed=0&scheduler=unipc-multi", + ), + TestCase( + "upscale-swinir-x4-2048-muffin", + "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-swinir-real-large-x4&scale=4&outscale=4", + source="txt2img-sd-v1-5-512-muffin-0", + ), + TestCase( + "upscale-codeformer-512-muffin", + "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&correction=correction-codeformer&faces=true&faceOutscale=1&faceStrength=1.0", + source="txt2img-sd-v1-5-512-muffin-0", + ), + TestCase( + "upscale-gfpgan-muffin", + "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=correction-gfpgan&faces=true&faceOutscale=1&faceStrength=1.0", + source="txt2img-sd-v1-5-512-muffin-0", + ), + TestCase( + "upscale-sd-x4-2048-muffin", + "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-stable-diffusion-x4&scale=4&outscale=4", + source="txt2img-sd-v1-5-512-muffin-0", + ), + TestCase( + "outpaint-panorama-even-256", + ( + "inpaint?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v1-inpainting&noise=fill-mask" + "&top=256&bottom=256&left=256&right=256&pipeline=panorama" + ), + source="txt2img-sd-v1-5-512-muffin-0", + mask="mask-black", + max_attempts=SLOW_TEST, + mse_threshold=0.025, + ), + TestCase( + "outpaint-panorama-vertical-512", + ( + "inpaint?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v1-inpainting&noise=histogram" + "&top=512&bottom=512&left=0&right=0&pipeline=panorama" + ), + source="txt2img-sd-v1-5-512-muffin-0", + mask="mask-black", + max_attempts=SLOW_TEST, + mse_threshold=0.025, + ), + TestCase( + "outpaint-panorama-horizontal-512", + ( + "inpaint?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v1-inpainting&noise=histogram" + "&top=0&bottom=0&left=512&right=512&pipeline=panorama" + ), + source="txt2img-sd-v1-5-512-muffin-0", + mask="mask-black", + max_attempts=SLOW_TEST, + mse_threshold=0.025, + ), ]