feat: ensure onnx option is tested

fboulnois · Feb 20, 2023 · 9fb71c6 · 9fb71c6
1 parent 908a9ca
commit 9fb71c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build.sh b/build.sh
@@ -51,7 +51,7 @@ tests() {
     TEST_IMAGE="An_impressionist_painting_of_a_parakeet_eating_spaghetti_in_the_desert_full.png"
     curl -sL "${BASE_URL}/${TEST_IMAGE}" > "$PWD/input/${TEST_IMAGE}"
     run --skip --height 512 --width 640 "abstract art"
-    run --device cpu --image "${TEST_IMAGE}" --strength 0.6 "abstract art"
+    run --device cpu --onnx --image "${TEST_IMAGE}" --strength 0.6 "abstract art"
     run --model "stabilityai/stable-diffusion-2" \
         --skip --height 768 --width 768 "abstract art"
     run --model "stabilityai/stable-diffusion-2-1" \