feat: add test for scheduler option

fboulnois · Dec 26, 2022 · a769727 · a769727
1 parent 22b7eda
commit a769727
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/build.sh b/build.sh
@@ -55,6 +55,7 @@ tests() {
         --prompt "An impressionist painting of a parakeet eating spaghetti in the desert"
     run --model "runwayml/stable-diffusion-v1-5" \
         --n_samples 2 --n_iter 2 --seed 42 \
+        --scheduler HeunDiscreteScheduler \
         --scale 7.5 --ddim_steps 80 --attention-slicing \
         --half --skip --negative-prompt "red roses" \
         --prompt "bouquet of roses"