apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(tests): update blend refs for release tests

This commit is contained in:
Sean Sube 2023-03-10 21:58:40 -06:00
parent cd11e8f3ab
commit 58250976d4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
api/scripts/test-refs/blend-512-muffin-blend-0.png (Stored with Git LFS)
View File

Binary file not shown.

BIN
api/scripts/test-refs/blend-512-muffin-white-0.png (Stored with Git LFS)
View File

Binary file not shown.

9
api/scripts/test-release.py
View File

@ -4,6 +4,7 @@ from io import BytesIO
from logging import getLogger from logging import getLogger
from logging.config import dictConfig from logging.config import dictConfig
from os import environ, path from os import environ, path
from sys import argv
from time import sleep from time import sleep
from typing import List, Optional, Union from typing import List, Optional, Union
@ -79,6 +80,10 @@ TEST_DATA = [
"txt2img-sd-v1-5-512-muffin-heun", "txt2img-sd-v1-5-512-muffin-heun",
"txt2img?prompt=a+giant+muffin&seed=0&scheduler=heun", "txt2img?prompt=a+giant+muffin&seed=0&scheduler=heun",
), ),
TestCase(
"txt2img-sd-v1-5-512-muffin-unipc",
"txt2img?prompt=a+giant+muffin&seed=0&scheduler=unipc-multi",
),
TestCase( TestCase(
"txt2img-sd-v2-1-512-muffin", "txt2img-sd-v2-1-512-muffin",
"txt2img?prompt=a+giant+muffin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v2-1", "txt2img?prompt=a+giant+muffin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v2-1",
@ -175,8 +180,8 @@ TEST_DATA = [
"blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim", "blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim",
mask="mask-white", mask="mask-white",
source=[ source=[
"txt2img-sd-v2-1-512-muffin-0",
"txt2img-sd-v1-5-512-muffin-0", "txt2img-sd-v1-5-512-muffin-0",
"txt2img-sd-v2-1-512-muffin-0",
], ],
), ),
TestCase( TestCase(
@ -184,8 +189,8 @@ TEST_DATA = [
"blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim", "blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim",
mask="mask-blend", mask="mask-blend",
source=[ source=[
"txt2img-sd-v2-1-512-muffin-0",
"txt2img-sd-v1-5-512-muffin-0", "txt2img-sd-v1-5-512-muffin-0",
"txt2img-sd-v2-1-512-muffin-0",
], ],
), ),
] ]