fix(tests): update blend refs for release tests

ssube · Mar 11, 2023 · 5825097 · 5825097
1 parent cd11e8f
commit 5825097
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/api/scripts/test-refs/blend-512-muffin-blend-0.png b/api/scripts/test-refs/blend-512-muffin-blend-0.png
diff --git a/api/scripts/test-refs/blend-512-muffin-white-0.png b/api/scripts/test-refs/blend-512-muffin-white-0.png
diff --git a/api/scripts/test-release.py b/api/scripts/test-release.py
@@ -4,6 +4,7 @@
 from logging import getLogger
 from logging.config import dictConfig
 from os import environ, path
+from sys import argv
 from time import sleep
 from typing import List, Optional, Union
 
@@ -79,6 +80,10 @@ def __init__(
         "txt2img-sd-v1-5-512-muffin-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(
         "txt2img-sd-v2-1-512-muffin",
         "txt2img?prompt=a+giant+muffin&seed=0&scheduler=ddim&model=stable-diffusion-onnx-v2-1",
@@ -175,17 +180,17 @@ def __init__(
         "blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim",
         mask="mask-white",
         source=[
-            "txt2img-sd-v2-1-512-muffin-0",
             "txt2img-sd-v1-5-512-muffin-0",
+            "txt2img-sd-v2-1-512-muffin-0",
         ],
     ),
     TestCase(
         "blend-512-muffin-blend",
         "blend?prompt=a+giant+pumpkin&seed=0&scheduler=ddim",
         mask="mask-blend",
         source=[
-            "txt2img-sd-v2-1-512-muffin-0",
             "txt2img-sd-v1-5-512-muffin-0",
+            "txt2img-sd-v2-1-512-muffin-0",
         ],
     ),
 ]