1
0
Fork 0

fix(api): add non-square txt2img test

This commit is contained in:
Sean Sube 2023-06-30 22:32:20 -05:00
parent 5378619ef2
commit fcaeb8cb42
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 16 additions and 0 deletions

BIN
api/scripts/test-refs/txt2img-sd-v1-5-tall-muffin-0.png (Stored with Git LFS) Normal file

Binary file not shown.

View File

@ -307,6 +307,19 @@ TEST_DATA = [
"img2img?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&sourceFilter=none&pipeline=panorama&tiledVAE=true", "img2img?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&sourceFilter=none&pipeline=panorama&tiledVAE=true",
source="txt2img-panorama-1024x768-muffin-0", source="txt2img-panorama-1024x768-muffin-0",
), ),
TestCase(
"txt2img-sd-v1-5-tall-muffin",
"txt2img?prompt=a+giant+muffin&seed=0&scheduler=ddim&width=512&height=768",
),
TestCase(
"upscale-resrgan-x4-tall-muffin",
(
"upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-real-esrgan-x4-plus&scale=4&outscale=4"
"&correction=correction-gfpgan&faces=false&faceOutscale=1&faceStrength=1.0"
),
source="txt2img-sd-v1-5-tall-muffin-0",
),
# TODO: non-square controlnet
] ]