From a769727acdffeed8cae984660343c91f1722dc6b Mon Sep 17 00:00:00 2001 From: fboulnois Date: Mon, 26 Dec 2022 13:45:01 -0500 Subject: [PATCH] feat: add test for scheduler option --- build.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/build.sh b/build.sh index f5541c7..7d8e530 100755 --- 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"